update dynamic shape (#41)

heheda12345 · May 17, 2024 · 6a048c7 · 6a048c7
2 parents 6ff29cf + 30045b3
commit 6a048c7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/frontend/fx_graph.py b/frontend/fx_graph.py
@@ -319,8 +319,8 @@ def compile(
                 for x in self.example_inputs
             ])
         assert callable(compiled_fn)
-        if self.fake_mode.shape_env is not None:
-            print("shape_env guards", self.fake_mode.shape_env.format_guards())
+        # if self.fake_mode.shape_env is not None:
+        #     print("shape_env guards", self.fake_mode.shape_env.format_guards())
         # TODO: add backend compiler
         return compiled_fn
 
@@ -402,9 +402,9 @@ def track_symint(source: StorePos, val: Any) -> None:
             if self.fake_mode.shape_env._maybe_evaluate_static(g) is not None:
                 print("maybe static")
                 continue
-            print("before simplify", g)
+            # print("before simplify", g)
             g = self.fake_mode.shape_env.simplify(g)
-            print("after simplify", g)
+            # print("after simplify", g)
             try:
                 codegen.add_check(
                     (ShapeGuardPrinter(symbol_to_source).doprint(g), voidpos()))