mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
8bdeb30510
This PR adds a flag --disable-check to the XGBoost benchmark script which disables the RuntimeError that comes up if training or prediction took too long. This is meant for non-CI exploratory use-cases. Specifically, the reason is this: We will include the XGBoost benchmark as an example workload for the KubeRay documentation. The actual performance of the workload is highly sensitive to infrastructure environment, so we won't want to raise an alarming RuntimeError if the workload took too long on the user's infrastructure. (When I tried the 100Gb benchmark on KubeRay, training ran just a couple of minutes longer than the 1000 second cutoff.) |
||
|---|---|---|
| .. | ||
| workloads | ||
| app_config.yaml | ||
| compute_cpu_1.yaml | ||
| compute_cpu_4.yaml | ||
| compute_cpu_8.yaml | ||
| compute_gpu_1.yaml | ||
| compute_gpu_2x2.yaml | ||
| compute_gpu_4_g4_12xl.yaml | ||
| compute_gpu_4x4.yaml | ||
| compute_gpu_8_g4_12xl.yaml | ||
| compute_gpu_16.yaml | ||
| data_20_nodes.yaml | ||
| xgboost_app_config.yaml | ||
| xgboost_compute_tpl.yaml | ||