修好了1.0

liqwang · Jan 11, 2024 · b3dbf65 · b3dbf65
1 parent 0543d71
commit b3dbf65
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/phase4/splc.py b/phase4/splc.py
@@ -61,9 +61,12 @@ def collect_params(ir_path) -> None:
 # the register allocation algorithm
 # 分配+查询reg
 def reg(var: str) -> str:
-    if var == '0':
-        return '$0'
-    else:
+    try:
+        if var == '0':
+            return '$0'
+        int(var)
+        return var
+    except:
         reg = find_reg(var)
         if reg:
             return reg
@@ -79,6 +82,7 @@ def reg(var: str) -> str:
         return f'{stack.index(var) + max_var_num << 2}($sp)'
 
 
+
 def find_reg(var: str):
     if var in register_table:
         return f'${register_table.index(var) + save_reg}'