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 2a654ea commit 5908dd5Copy full SHA for 5908dd5
onnx_diagnostic/torch_export_patches/patches/patch_transformers.py
@@ -51,10 +51,10 @@ def vector_mask_function(
51
for a, dims in zip(args, udimensions)
52
]
53
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)
+ # if is_torchdynamo_exporting():
+ # for a in args:
+ # # The exporter should export with a dimension > 1 to make sure it is dynamic.
+ # torch._check(a.shape[0] > 1)
58
expanded_args = [a.expand(max_shape) for a in new_args]
59
return mask_function(*expanded_args)
60
0 commit comments