Merge pull request #16 from intelligent-machine-learning/hotfix/fix_X…

…laDeviceType fix torch_xla/csrc/aten_xla_type.cpp
intelligent-machine-learning · Apr 2, 2024 · 123ad3b · 123ad3b
2 parents 4b6b78d + 663a984
commit 123ad3b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_xla/csrc/aten_xla_type.cpp b/torch_xla/csrc/aten_xla_type.cpp
@@ -3237,7 +3237,7 @@ at::Tensor XLANativeFunctions::upsample_bicubic2d(
   XlaDeviceType hw_type =
       static_cast<XlaDeviceType>(self_tensor->GetDevice().type());
   // NOT GPU
-  if (hw_type != XlaDeviceType::CUDA && hw_type != XlaDeviceType::GPU) {
+  if (hw_type != XlaDeviceType::CUDA) {
     return at::native::call_fallback_fn<
         &xla_cpu_fallback, ATEN_OP(upsample_bicubic2d)>::call(self, output_size,
                                                               align_corners,
@@ -3271,7 +3271,7 @@ at::Tensor XLANativeFunctions::upsample_bicubic2d_backward(
   // NOT GPU
   XlaDeviceType hw_type =
       static_cast<XlaDeviceType>(grad_output_tensor->GetDevice().type());
-  if (hw_type != XlaDeviceType::CUDA && hw_type != XlaDeviceType::GPU) {
+  if (hw_type != XlaDeviceType::CUDA) {
     return at::native::call_fallback_fn<
         &xla_cpu_fallback,
         ATEN_OP(upsample_nearest2d_backward)>::call(grad_output, output_size,