X-Git-Url: https://git.libre-soc.org/?a=blobdiff_plain;f=src%2Farch%2Fmicro_asm_test.py;h=b074ecb58f55e20110102f25b06bdadb70ece2cb;hb=5836023ab2eb8f2c780e835078d518f061722d65;hp=858ac511eedb327bb27af2ab9bf1fdc255656740;hpb=be0aef981943d123e0f4eb8c5520a74c1eb1002a;p=gem5.git

diff --git a/src/arch/micro_asm_test.py b/src/arch/micro_asm_test.py
index 858ac511e..b074ecb58 100755
--- a/src/arch/micro_asm_test.py
+++ b/src/arch/micro_asm_test.py
@@ -26,7 +26,7 @@
 #
 # Authors: Gabe Black
 
-from micro_asm import MicroAssembler, Macroop, Rom
+from micro_asm import MicroAssembler, Combinational_Macroop, Rom_Macroop, Rom
 
 class Bah(object):
     def __init__(self):
@@ -52,7 +52,7 @@ microops = {
     "dah": Dah
 }
 
-class TestMacroop(Macroop):
+class TestMacroop(Combinational_Macroop):
     def tweak(self):
         microops["bah"] = Bah_Tweaked
     def untweak(self):
@@ -68,7 +68,7 @@ class TestMacroop(Macroop):
             "print": self.print_debug
         }
 
-assembler = MicroAssembler(TestMacroop, microops, Rom('main ROM'))
+assembler = MicroAssembler(TestMacroop, microops, Rom('main ROM'), Rom_Macroop)
 
 testAssembly = '''
 # Single line comment
@@ -91,6 +91,13 @@ def macroop squishy {
     .tweak
 };
 
+#Extending the rom...
+def rom
+{
+    #Here's more stuff for the rom
+    bah
+};
+
 def macroop squashy {
     bah
 };