From 2cb2fce8d905a014ed9b1d53618ab04aae918ef4 Mon Sep 17 00:00:00 2001 From: chenyu Date: Wed, 5 Mar 2025 14:06:11 -0500 Subject: [PATCH] lower test_gemm_8192 amd_tflops to 65 (#9364) --- test/external/speed_v_theoretical.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/external/speed_v_theoretical.py b/test/external/speed_v_theoretical.py index b365b471e4..724a2ad517 100644 --- a/test/external/speed_v_theoretical.py +++ b/test/external/speed_v_theoretical.py @@ -91,7 +91,7 @@ class TestKernelSpeed(unittest.TestCase): # theoretical is nv_tflops=165, amd_tflops=123 def test_gemm_4096(self): self._test_matmul(4096, nv_tflops=115, amd_tflops=80) - def test_gemm_8192(self): self._test_matmul(8192, nv_tflops=125, amd_tflops=70) + def test_gemm_8192(self): self._test_matmul(8192, nv_tflops=125, amd_tflops=65) # theoretical is nv_gbs=1008, amd_gbs=960 def test_gemv_16384_4096(self): self._test_matmul(16384, 4096, 1, nv_gbs=840, amd_gbs=750)