14
14
# Faster crates.io index checkout.
15
15
CARGO_REGISTRIES_CRATES_IO_PROTOCOL : sparse
16
16
RUST_LOG : debug
17
+ # Build the kernel only for the single architecture . This should reduce
18
+ # the overall compile-time significantly.
19
+ EC_GPU_CUDA_NVCC_ARGS : --fatbin --gpu-architecture=sm_75 --generate-code=arch=compute_75,code=sm_75
20
+ BELLMAN_CUDA_NVCC_ARGS : --fatbin --gpu-architecture=sm_75 --generate-code=arch=compute_75,code=sm_75
21
+ NEPTUNE_CUDA_NVCC_ARGS : --fatbin --gpu-architecture=sm_75 --generate-code=arch=compute_75,code=sm_75
17
22
18
23
jobs :
19
24
check_clippy :
20
25
runs-on : ubuntu-24.04
21
26
name : Clippy
22
- env :
23
- # Build the kernel only for the single architecture . This should reduce
24
- # the overall compile-time significantly.
25
- EC_GPU_CUDA_NVCC_ARGS : --fatbin --gpu-architecture=sm_75 --generate-code=arch=compute_75,code=sm_75
26
- BELLMAN_CUDA_NVCC_ARGS : --fatbin --gpu-architecture=sm_75 --generate-code=arch=compute_75,code=sm_75
27
- NEPTUNE_CUDA_NVCC_ARGS : --fatbin --gpu-architecture=sm_75 --generate-code=arch=compute_75,code=sm_75
28
27
steps :
29
28
- uses : actions/checkout@v4
30
29
- name : Install required packages
@@ -40,45 +39,14 @@ jobs:
40
39
- name : Run cargo fmt
41
40
run : cargo fmt --all -- --check
42
41
43
- build_gpu :
42
+ test :
44
43
runs-on : ubuntu-24.04
45
- name : Build
46
- env :
47
- RUST_LOG : trace
48
- # Build the kernel only for the single architecture . This should reduce
49
- # the overall compile-time significantly.
50
- EC_GPU_CUDA_NVCC_ARGS : --fatbin --gpu-architecture=sm_75 --generate-code=arch=compute_75,code=sm_75
51
- BELLMAN_CUDA_NVCC_ARGS : --fatbin --gpu-architecture=sm_75 --generate-code=arch=compute_75,code=sm_75
52
- NEPTUNE_CUDA_NVCC_ARGS : --fatbin --gpu-architecture=sm_75 --generate-code=arch=compute_75,code=sm_75
44
+ name : Test
53
45
steps :
54
46
- uses : actions/checkout@v4
55
47
- name : Install required packages
56
48
run : sudo apt install --no-install-recommends --yes libhwloc-dev nvidia-cuda-toolkit ocl-icd-opencl-dev
57
- # # Run clippy as apart of the test job as it also needs the nvidia toolkit
58
- # # in order to compile
59
- # - name: Run Clippy
60
- # run: cargo clippy --workspace -- -D warnings
61
- - name : Build it
62
- run : cargo build --verbose
63
-
64
- test_gpu :
65
- runs-on : ubuntu-24.04
66
- name : Test on GPUs
67
- env :
68
- RUST_LOG : trace
69
- # Build the kernel only for the single architecture . This should reduce
70
- # the overall compile-time significantly.
71
- EC_GPU_CUDA_NVCC_ARGS : --fatbin --gpu-architecture=sm_75 --generate-code=arch=compute_75,code=sm_75
72
- BELLMAN_CUDA_NVCC_ARGS : --fatbin --gpu-architecture=sm_75 --generate-code=arch=compute_75,code=sm_75
73
- NEPTUNE_CUDA_NVCC_ARGS : --fatbin --gpu-architecture=sm_75 --generate-code=arch=compute_75,code=sm_75
74
- steps :
75
- - uses : actions/checkout@v4
76
- - name : Install required packages
77
- run : sudo apt install --no-install-recommends --yes libhwloc-dev nvidia-cuda-toolkit ocl-icd-opencl-dev
78
- # Run clippy as apart of the test job as it also needs the nvidia toolkit
79
- # in order to compile
80
- - name : Run Clippy
81
- run : cargo clippy --workspace -- -D warnings
49
+ # In case no GPUs are available, it's using the CPU fallback.
82
50
- name : Test with CUDA
83
51
run : cargo test --verbose
84
52
0 commit comments