Skip to content

Commit 5be0baf

Browse files
author
Tristan Konolige
committed
lower flops ratio, machine may be using multiple threads
1 parent 4be1d83 commit 5be0baf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/python/unittest/test_runtime_profiling.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -267,8 +267,8 @@ def test_estimate_peak_fma_flops(target, dev):
267267
flops = tvm.utils.estimate_peak_fma_flops(tvm.target.Target(target), dev)
268268
# Assume we can achieve 1 GFLOP/s per thread, which is 1 FLOP per cycle on a 1GHz cpu.
269269
assert (
270-
flops > 10**9 * tvm.runtime.num_threads() and flops < 10**14
271-
), f"FLOP/s should be between 10^9 * num_threads and 10^14, but it is {flops}"
270+
flops > 10**9 and flops < 10**14
271+
), f"FLOP/s should be between 10^9 and 10^14, but it is {flops}"
272272

273273

274274
def test_estimate_peak_fma_flops_rpc():

0 commit comments

Comments
 (0)