Skip to content

Commit c341d98

Browse files
authored
fix: example argument issue raised in 4070 (#4071)
1 parent 84ef25b commit c341d98

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

examples/dynamo/torch_compile_resnet_example.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@
4848
model,
4949
ir="torch_compile",
5050
inputs=inputs,
51+
use_explicit_typing=False,
5152
enabled_precisions=enabled_precisions,
5253
workspace_size=workspace_size,
5354
min_block_size=min_block_size,
@@ -86,6 +87,7 @@
8687
model,
8788
ir="torch_compile",
8889
inputs=inputs_bs8,
90+
use_explicit_typing=False,
8991
enabled_precisions=enabled_precisions,
9092
workspace_size=workspace_size,
9193
min_block_size=min_block_size,
@@ -111,6 +113,7 @@
111113
dtype=torch.half,
112114
)
113115
],
116+
"use_explicit_typing": False,
114117
"enabled_precisions": enabled_precisions,
115118
"ir": "dynamo",
116119
}

0 commit comments

Comments
 (0)