Anton Gusev 5 роки тому
батько
коміт
0adadf649f
1 змінених файлів з 1 додано та 0 видалено
  1. 1 0
      tests/benchmark_throughput.py

+ 1 - 0
tests/benchmark_throughput.py

@@ -110,6 +110,7 @@ def benchmark_throughput(num_experts=16, num_handlers=None, num_clients=128, num
         print("Note: benchmark code failed, timing/memory results only indicate time till failure!")
     print_device_info(device)
     print(flush=True)
+    assert not benchmarking_failed.is_set()
 
 
 if __name__ == "__main__":