apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 16:48:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Test] scaled_gemm_f8_i4_f16 skip test while sm != 89 (#3210)

Browse Source
This commit is contained in:
ming1753
2025-08-05 15:25:28 +08:00
committed by GitHub
parent 40f7f3e0d8
commit 14ed75f7d3
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/operators/test_scaled_gemm_f8_i4_f16.py
View File

@@ -33,6 +33,10 @@ class Test(unittest.TestCase):
paddle.seed(2024) paddle.seed(2024)
print(paddle.device.cuda.get_device_properties()) print(paddle.device.cuda.get_device_properties())
print(paddle.__git_commit__) print(paddle.__git_commit__)
prop = paddle.device.cuda.get_device_properties()
cc = prop.major * 10 + prop.minor
if cc != 89:
self.skipTest("scaled_gemm_f8_i4_f16 only support sm 89!")
def quant_fp8_pertensor(self, tensor): def quant_fp8_pertensor(self, tensor):
""" """