Generate also jax-cuda-plugin and jax-cuda-pjrt in cuda builds and bump CUDA used at built time to 12.6 and add fixes for CUDA #108
Job | Run time |
---|---|
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
12s |
Job | Run time |
---|---|
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
1s | |
12s |