From 1f4ce1aebf7d4ade55d6c2153dac4387f2469790 Mon Sep 17 00:00:00 2001
From: li_jinxin <lijinxin6@huawei.com>
Date: Thu, 7 Dec 2023 16:29:32 +0800
Subject: [PATCH] add skip if unsupport device.

---
 torch_npu/testing/common_distributed.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/torch_npu/testing/common_distributed.py b/torch_npu/testing/common_distributed.py
index c19d3ff033..9a55c2f75f 100644
--- a/torch_npu/testing/common_distributed.py
+++ b/torch_npu/testing/common_distributed.py
@@ -27,3 +27,14 @@ def skipIfUnsupportMultiNPU(npu_number_needed):
             return func(self)
         return wrapper
     return skip_dec
+
+
+def skipOpOnlySupport910B(op_name):
+    def skip_dec(func):
+        def wrapper(self):
+            device_name = torch_npu.npu.get_device_name(0)[:10]
+            if device_name != 'Ascend910B':
+                raise unittest.SkipTest(f"The op {op_name} is only suppport on 910B, skip this ut for this device type!")
+            return func(self)
+        return wrapper
+    return skip_dec
-- 
Gitee