add XLA backend into tensor type strings (#86881)

add XLA backend into tensor type strings Pull Request resolved: #86881 Approved by: https://github.com/bdhirsh
pytorch · Oct 17, 2022 · 33343de · 33343de
1 parent 317eeb8
commit 33343de
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torch/csrc/utils/tensor_types.cpp b/torch/csrc/utils/tensor_types.cpp
@@ -45,6 +45,8 @@ static const char* backend_to_string(const at::Backend& backend) {
       return "torch.privateuseone";
     case at::Backend::Lazy:
       return "torch.lazy";
+    case at::Backend::XLA:
+      return "torch.xla";
     default:
       AT_ERROR("Unimplemented backend ", backend);
   }