Skip to content

Commit 6a048c7

Browse files
authored
update dynamic shape (apache#41)
2 parents 6ff29cf + 30045b3 commit 6a048c7

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

frontend/fx_graph.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -319,8 +319,8 @@ def compile(
319319
for x in self.example_inputs
320320
])
321321
assert callable(compiled_fn)
322-
if self.fake_mode.shape_env is not None:
323-
print("shape_env guards", self.fake_mode.shape_env.format_guards())
322+
# if self.fake_mode.shape_env is not None:
323+
# print("shape_env guards", self.fake_mode.shape_env.format_guards())
324324
# TODO: add backend compiler
325325
return compiled_fn
326326

@@ -402,9 +402,9 @@ def track_symint(source: StorePos, val: Any) -> None:
402402
if self.fake_mode.shape_env._maybe_evaluate_static(g) is not None:
403403
print("maybe static")
404404
continue
405-
print("before simplify", g)
405+
# print("before simplify", g)
406406
g = self.fake_mode.shape_env.simplify(g)
407-
print("after simplify", g)
407+
# print("after simplify", g)
408408
try:
409409
codegen.add_check(
410410
(ShapeGuardPrinter(symbol_to_source).doprint(g), voidpos()))

0 commit comments

Comments
 (0)