Anton Gusev 5 жил өмнө
parent
commit
0adadf649f

+ 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__":