rapidsai / crossfit

Metric calculation library
Apache License 2.0
2 stars 5 forks source link

Fix CI #60

Closed VibhuJawa closed 2 months ago

VibhuJawa commented 2 months ago

This PR enables GPU CI on this repo:

Tested locally using:

docker run --gpus all -v $(pwd):/workspace/crossfit --rm -it rapidsai/base:24.06-cuda12.2-py3.11  bash -c "cd /workspace/crossfit && bash ci/test_gpu.sh"
VibhuJawa commented 2 months ago

CI seems to be passing again .

CC: @marcromeyn / @edknv for quick eyes on this.