We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e6708c9 commit 78b56b4Copy full SHA for 78b56b4
tests/py/dynamo/conversion/harness.py
@@ -368,6 +368,12 @@ def generate_graph(
368
tuple(torch_export_inputs),
369
dynamic_shapes=torch_export_dynamic_shapes,
370
)
371
+ # for the dynamo tracer, with dynamic shapes,
372
+ # we need to apply the post lowering passes especially remove _assert_scalar nodes
373
+ for dynamic_shape_spec in torch_export_dynamic_shapes.values():
374
+ if len(dynamic_shape_spec) > 0:
375
+ enable_passes = True
376
+ break
377
if enable_passes:
378
exported_program = pre_export_lowering(exported_program, settings)
379
exported_program = exported_program.run_decompositions(
0 commit comments