Merge branch 'master' into update_fuse_layer_norm_grad_for_npu

Oneflow-Inc · Jan 22, 2025 · 9a72438 · 9a72438
2 parents ce66dff + 5c62322
commit 9a72438
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/oneflow/core/common/device_type.proto b/oneflow/core/common/device_type.proto
@@ -9,4 +9,5 @@ enum DeviceType {
   kMeta = 4;
   kMLU = 5;  // Cambricon MLU
   kNPU = 6;  // Ascend NPU
+  kXPU = 7;  // KunLunXin
 }
diff --git a/oneflow/ir/lib/OneFlow/UserOpConversion.cpp b/oneflow/ir/lib/OneFlow/UserOpConversion.cpp
@@ -488,6 +488,8 @@ ::oneflow::DeviceType getDeviceTypeFromAttrDictionary(DictionaryAttr attributes)
     return ::oneflow::DeviceType::kMLU;
   } else if (device_tag.str() == "npu") {
     return ::oneflow::DeviceType::kNPU;
+  } else if (device_tag.str() == "xpu") {
+    return ::oneflow::DeviceType::kXPU;
   } else {
     LOG(FATAL) << "unsupported device tag: " << device_tag.str();
     return ::oneflow::DeviceType::kInvalidDevice;