未验证 提交 30674b0e 编写于 作者: Y YUNSHEN XIE 提交者: GitHub

add condition of skipif (#48791)

* add condition of skipif

* fix code format error

* Update test_fused_gate_attention_op.py

update
上级 6c9df13d
...@@ -402,7 +402,9 @@ class TestMergeQKVLargeBatchSizeFp16Case(TestMergeQKVFp16Case): ...@@ -402,7 +402,9 @@ class TestMergeQKVLargeBatchSizeFp16Case(TestMergeQKVFp16Case):
@unittest.skipIf( @unittest.skipIf(
not core.is_compiled_with_cuda() or get_cuda_version() < 11000, not core.is_compiled_with_cuda()
or get_cuda_version() < 11000
or paddle.device.cuda.get_device_capability()[0] < 8,
"core is not compiled with CUDA and cuda version need larger than or equal to 11.3", "core is not compiled with CUDA and cuda version need larger than or equal to 11.3",
) )
class TestMergeQKVBF16Case(TestFusedGateAttentionOp): class TestMergeQKVBF16Case(TestFusedGateAttentionOp):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册