Skip to content

Commit 5908dd5

Browse files
committed
fix issues
1 parent 2a654ea commit 5908dd5

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

onnx_diagnostic/torch_export_patches/patches/patch_transformers.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,10 @@ def vector_mask_function(
5151
for a, dims in zip(args, udimensions)
5252
]
5353
max_shape = tuple(args[i].shape[0] for i in indices)
54-
if is_torchdynamo_exporting():
55-
for a in args:
56-
# The exporter should export with a dimension > 1 to make sure it is dynamic.
57-
torch._check(a.shape[0] > 1)
54+
# if is_torchdynamo_exporting():
55+
# for a in args:
56+
# # The exporter should export with a dimension > 1 to make sure it is dynamic.
57+
# torch._check(a.shape[0] > 1)
5858
expanded_args = [a.expand(max_shape) for a in new_args]
5959
return mask_function(*expanded_args)
6060

0 commit comments

Comments
 (0)