From 9e56681c7191f8c96cf09ff399edfb5dc58a115b Mon Sep 17 00:00:00 2001
From: liujingfeng4A069 <liujingfeng@sensetime.com>
Date: Fri, 2 Aug 2024 14:47:19 +0800
Subject: [PATCH] merge the newest main branch

---
 impl/ascend/device_configs.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/impl/ascend/device_configs.py b/impl/ascend/device_configs.py
index 1377c420e..1997c29b1 100755
--- a/impl/ascend/device_configs.py
+++ b/impl/ascend/device_configs.py
@@ -1214,10 +1214,10 @@
     ),
     
     # TODO(zhangqiu) Due to a bug in the software stack, this test will be skipped for now.
-    'token_attention': dict(
-        name=['token_attention'],
-        skip_all=True
-    ),
+    # 'token_attention': dict(
+    #     name=['token_attention'],
+    #     skip_all=True
+    # ),
 
     'rotary_emb': dict(
         name=["rotary_emb"],
@@ -1244,10 +1244,10 @@
     # ),
     
     # TODO(zhangqiu) Due to a bug in the software stack, this test will be skipped for now.
-    'token_softmax_reducev': dict(
-        name=['token_softmax_reducev'],
-        skip_all=True
-    ),
+    # 'token_softmax_reducev': dict(
+    #     name=['token_softmax_reducev'],
+    #     skip_all=True
+    # ),
 
     # temp for 910B
     'normal_': dict(