* Update to C++17 * Turn off unity build * Update CMake to 3.18 * Use MSVC 2022 + CUDA 11.8 * Re-create stack for worker images * Allocate more disk space for Windows * Tempiorarily disable clang-tidy * RAPIDS now requires Python 3.10+ * Unpin cuda-python * Use latest NCCL * Use Ubuntu 20.04 in RMM image * Mark failing mgpu test as xfail
19 lines
656 B
Python
19 lines
656 B
Python
IMAGE_PARAMS = {
|
|
"linux-amd64-gpu": {
|
|
"BaseImageId": "linuxamd64",
|
|
# AMI ID is looked up from Buildkite's CloudFormation template
|
|
"BootstrapScript": "linux-amd64-gpu-bootstrap.yml",
|
|
"InstanceType": "g4dn.xlarge",
|
|
"InstanceOperatingSystem": "Linux",
|
|
"VolumeSize": "40", # in GiBs
|
|
},
|
|
"windows-gpu": {
|
|
"BaseImageId": "windows",
|
|
# AMI ID is looked up from Buildkite's CloudFormation template
|
|
"BootstrapScript": "windows-gpu-bootstrap.yml",
|
|
"InstanceType": "g4dn.2xlarge",
|
|
"InstanceOperatingSystem": "Windows",
|
|
"VolumeSize": "120", # in GiBs
|
|
},
|
|
}
|