From a7ddbe96ef5ced594f1e98e3a83842a78a6ea478 Mon Sep 17 00:00:00 2001 From: Philipp van Kempen Date: Mon, 23 Oct 2023 13:31:52 +0200 Subject: [PATCH] [Tests] Check int8+int32 testcases in test_estimate_peak_flops_cpu Pytest param fixture for dtypes was not used resulting in these two test cases not beeing checked. --- tests/python/unittest/test_roofline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/unittest/test_roofline.py b/tests/python/unittest/test_roofline.py index 7a1c3478c51d..cb8336630e60 100644 --- a/tests/python/unittest/test_roofline.py +++ b/tests/python/unittest/test_roofline.py @@ -42,7 +42,7 @@ def test_estimate_peak_flops_cpu(dtype): target = tvm.target.Target("llvm -mattr=+fma,+avx2") dev = remote.device(str(target)) # This test uses vectorized instructions so we need a target that supports them - flops = tvm.utils.roofline.x86.estimate_peak_fma_vector_flops(target, dev, remote, "float32") + flops = tvm.utils.roofline.x86.estimate_peak_fma_vector_flops(target, dev, remote, dtype) # Assume we can achieve 1 GFLOP/s per thread, which is 1 FLOP per cycle on a 1GHz cpu. assert ( flops > 10**9 and flops < 10**14