Fix torch.full scalar type

pytorch · May 1, 2024 · 8b1ca55 · 8b1ca55
1 parent 2907ab3
commit 8b1ca55
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_xla/csrc/aten_xla_type.cpp b/torch_xla/csrc/aten_xla_type.cpp
@@ -1472,7 +1472,7 @@ at::Tensor XLANativeFunctions::full(at::IntArrayRef size,
   }
   return bridge::AtenFromXlaTensor(tensor_methods::full(
       absl::Span<const int64_t>(size), fill_value,
-      GetXlaDeviceOrCurrent(device), at::dtype_or_default(dtype)));
+      GetXlaDeviceOrCurrent(device), dtype ? *dtype : fill_value.type()));
 }
 
 at::Tensor XLANativeFunctions::gather(const at::Tensor& self, int64_t dim,