瀏覽代碼

add not failed assertion

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