Open vmx opened 1 year ago
Currently only CI only the the tree building is run on the GPU. It might make sense to also run other tests on GPUs.
Other tests run on GPU
Getting GPUs working correctly on CI can be tricky. Though we already have a gpu executor, so it might be not too hard.
Try to run the test_release job on a GPU instance with CUDA enabled.
test_release
Description
Currently only CI only the the tree building is run on the GPU. It might make sense to also run other tests on GPUs.
Acceptance criteria
Other tests run on GPU
Risks + pitfalls
Getting GPUs working correctly on CI can be tricky. Though we already have a gpu executor, so it might be not too hard.
Where to begin
Try to run the
test_release
job on a GPU instance with CUDA enabled.