Skip to content

Commit b9bef19

Browse files
fixed the type miss match issue with symint in SPMD regime and dynamo (#8681)
1 parent 49cd310 commit b9bef19

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch_xla/_dynamo/dynamo_bridge.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -581,7 +581,7 @@ def optimized_mod(*args: tuple):
581581
# we will skip checking the input sharding since it can be expensive.
582582
if skip_checking_input_sharding_threashold > 0:
583583
if torch_xla._XLAC._get_xla_sharding_specs(
584-
args) != xla_args_sharding_spec:
584+
xla_args_tensor_only) != xla_args_sharding_spec:
585585
# update the xla_args with the input with new sharding and retrace
586586
xla_model.xla_args = args
587587
(xla_args_sharding_spec, args_and_out_copy, graph_hash,

0 commit comments

Comments
 (0)