We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5ac2d1a commit c4c0a49Copy full SHA for c4c0a49
1 file changed
tests/python/unittest/test_runtime_module_based_interface.py
@@ -680,9 +680,9 @@ def test_num_threads():
680
env_threads = os.getenv("TVM_NUM_THREADS")
681
omp_env_threads = os.getenv("OMP_NUM_THREADS")
682
if env_threads is not None:
683
- assert reported == env_threads
+ assert reported == int(env_threads)
684
elif omp_env_threads is not None:
685
- assert reported == omp_env_threads
+ assert reported == int(omp_env_threads)
686
else:
687
hardware_threads = os.cpu_count()
688
assert reported == hardware_threads or reported == hardware_threads // 2
0 commit comments