Skip to content

Commit 14ed75f

Browse files
authored
[Test] scaled_gemm_f8_i4_f16 skip test while sm != 89 (#3210)
1 parent 40f7f3e commit 14ed75f

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

test/operators/test_scaled_gemm_f8_i4_f16.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,10 @@ def setUp(self):
3333
paddle.seed(2024)
3434
print(paddle.device.cuda.get_device_properties())
3535
print(paddle.__git_commit__)
36+
prop = paddle.device.cuda.get_device_properties()
37+
cc = prop.major * 10 + prop.minor
38+
if cc != 89:
39+
self.skipTest("scaled_gemm_f8_i4_f16 only support sm 89!")
3640

3741
def quant_fp8_pertensor(self, tensor):
3842
"""

0 commit comments

Comments
 (0)