From c9a2b9c7d414109173b30f2672daa8b2f7334080 Mon Sep 17 00:00:00 2001 From: Jason Ansel Date: Mon, 11 Jul 2022 14:37:59 -0700 Subject: [PATCH] [FRONTEND] Add missing args to get_simd_tflops() (#578) --- python/triton/ops/matmul_perf_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/triton/ops/matmul_perf_model.py b/python/triton/ops/matmul_perf_model.py index 9c10b88d8..004f236b9 100644 --- a/python/triton/ops/matmul_perf_model.py +++ b/python/triton/ops/matmul_perf_model.py @@ -26,7 +26,7 @@ def get_simd_tflops(backend, device, num_ctas, num_warps, dtype): def get_tflops(backend, device, num_ctas, num_warps, dtype): cc = _triton.runtime.cc(backend, device) if cc < 80 and dtype == torch.float32: - return get_simd_tflops() + return get_simd_tflops(backend, device, num_ctas, num_warps, dtype) return get_tensorcore_tflops(backend, device, num_ctas, num_warps, dtype)