未验证 提交 70554c9f 编写于 作者: Y YUNSHEN XIE 提交者: GitHub

disable TestSoftmaxFP16Op2 in test_softmax_op (#25519)

上级 e4017e5c
...@@ -181,8 +181,7 @@ class TestSoftmaxFP16Op(TestSoftmaxOp): ...@@ -181,8 +181,7 @@ class TestSoftmaxFP16Op(TestSoftmaxOp):
pass pass
@unittest.skipIf(not core.is_compiled_with_cuda(), @unittest.skip('disable TestSoftmaxFP16Op2')
"core is not compiled with CUDA")
class TestSoftmaxFP16Op2(TestSoftmaxOp): class TestSoftmaxFP16Op2(TestSoftmaxOp):
def init_kernel_type(self): def init_kernel_type(self):
self.dtype = np.float16 self.dtype = np.float16
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册