Dont wrap ints in torch.tensor;

as it tickles the bug #6538
pytorch · Feb 15, 2024 · e4d8643 · e4d8643
1 parent 29b01c6
commit e4d8643
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_xla/stablehlo.py b/torch_xla/stablehlo.py
@@ -248,7 +248,7 @@ def call_function(self, target, args: Tuple, kwargs: Dict) -> Any:
         args = tuple(
             map(
                 lambda arg_spec: torch.tensor(arg_spec[0])
-                if isinstance(arg_spec[0], (float, int)) and type(arg_spec[
+                if isinstance(arg_spec[0], float) and type(arg_spec[
                     1].type) == torch.TensorType else arg_spec[0],
                 args_and_specs))
     return super().call_function(target, args, new_kwargs)