Fix quantile tests running on multi-gpus (#8775)
* Fix quantile tests running on multi-gpus * Run some gtests with multiple GPUs * fix mgpu test naming * Instruct NCCL to print extra logs * Allocate extra space in /dev/shm to enable NCCL * use gtest_skip to skip mgpu tests --------- Co-authored-by: Hyunsu Philip Cho <chohyu01@cs.washington.edu>
This commit is contained in:
@@ -36,6 +36,11 @@ steps:
|
||||
queue: linux-amd64-mgpu
|
||||
- wait
|
||||
#### -------- TEST --------
|
||||
- label: ":console: Run Google Tests"
|
||||
command: "tests/buildkite/test-cpp-mgpu.sh"
|
||||
key: test-cpp-mgpu
|
||||
agents:
|
||||
queue: linux-amd64-mgpu
|
||||
- label: ":console: Test Python package, 4 GPUs"
|
||||
command: "tests/buildkite/test-python-gpu.sh mgpu"
|
||||
key: test-python-mgpu
|
||||
|
||||
Reference in New Issue
Block a user