From 2fcfbe335ebfe987cde76e892d4c4056ccace177 Mon Sep 17 00:00:00 2001
From: huangyunlong <huangyunlong4@huawei.com>
Date: Wed, 1 Nov 2023 16:30:55 +0800
Subject: [PATCH] [Feature]support new socversion

---
 torch_npu/csrc/core/npu/NpuVariables.cpp | 1 +
 torch_npu/csrc/core/npu/NpuVariables.h   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/torch_npu/csrc/core/npu/NpuVariables.cpp b/torch_npu/csrc/core/npu/NpuVariables.cpp
index d33962f9f2..84746b06e0 100644
--- a/torch_npu/csrc/core/npu/NpuVariables.cpp
+++ b/torch_npu/csrc/core/npu/NpuVariables.cpp
@@ -38,6 +38,7 @@ static std::map<std::string, SocVersion> socVersionMap = {
     {"Ascend310P4", SocVersion::Ascend310P4},
     {"Ascend910B1", SocVersion::Ascend910B1},
     {"Ascend910B2", SocVersion::Ascend910B2},
+    {"Ascend910B2C", SocVersion::Ascend910B2C},
     {"Ascend910B3", SocVersion::Ascend910B3},
     {"Ascend910B4", SocVersion::Ascend910B4},
     {"Ascend310B1", SocVersion::Ascend310B1},
diff --git a/torch_npu/csrc/core/npu/NpuVariables.h b/torch_npu/csrc/core/npu/NpuVariables.h
index 527e71c935..82b6138320 100644
--- a/torch_npu/csrc/core/npu/NpuVariables.h
+++ b/torch_npu/csrc/core/npu/NpuVariables.h
@@ -30,6 +30,7 @@ enum class SocVersion {
   Ascend310P4,
   Ascend910B1 = 220,
   Ascend910B2,
+  Ascend910B2C,
   Ascend910B3,
   Ascend910B4,
   Ascend310B1 = 240,
-- 
Gitee