Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Generate also jax-cuda-plugin and jax-cuda-pjrt in cuda builds and bump CUDA used at built time to 12.6 #288

Open
wants to merge 9 commits into
base: main
Choose a base branch
from
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ c_stdlib:
c_stdlib_version:
- '2.17'
cdt_name:
- cos7
- conda
channel_sources:
- conda-forge
channel_targets:
Expand Down Expand Up @@ -50,7 +50,6 @@ zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cdt_name
- cuda_compiler
- cuda_compiler_version
- docker_image
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ c_stdlib:
c_stdlib_version:
- '2.17'
cdt_name:
- cos7
- conda
channel_sources:
- conda-forge
channel_targets:
Expand Down Expand Up @@ -50,7 +50,6 @@ zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cdt_name
- cuda_compiler
- cuda_compiler_version
- docker_image
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ c_stdlib:
c_stdlib_version:
- '2.17'
cdt_name:
- cos7
- conda
channel_sources:
- conda-forge
channel_targets:
Expand Down Expand Up @@ -50,7 +50,6 @@ zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cdt_name
- cuda_compiler
- cuda_compiler_version
- docker_image
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ c_stdlib:
c_stdlib_version:
- '2.17'
cdt_name:
- cos7
- conda
channel_sources:
- conda-forge
channel_targets:
Expand Down Expand Up @@ -50,7 +50,6 @@ zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cdt_name
- cuda_compiler
- cuda_compiler_version
- docker_image
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ c_stdlib:
c_stdlib_version:
- '2.17'
cdt_name:
- cos7
- conda
channel_sources:
- conda-forge
channel_targets:
Expand Down Expand Up @@ -50,7 +50,6 @@ zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cdt_name
- cuda_compiler
- cuda_compiler_version
- docker_image
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ c_stdlib:
c_stdlib_version:
- '2.17'
cdt_name:
- cos7
- conda
channel_sources:
- conda-forge
channel_targets:
Expand Down Expand Up @@ -50,7 +50,6 @@ zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cdt_name
- cuda_compiler
- cuda_compiler_version
- docker_image
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ c_stdlib:
c_stdlib_version:
- '2.17'
cdt_name:
- cos7
- conda
channel_sources:
- conda-forge
channel_targets:
Expand Down Expand Up @@ -50,7 +50,6 @@ zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cdt_name
- cuda_compiler
- cuda_compiler_version
- docker_image
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ c_stdlib:
c_stdlib_version:
- '2.17'
cdt_name:
- cos7
- conda
channel_sources:
- conda-forge
channel_targets:
Expand Down Expand Up @@ -50,7 +50,6 @@ zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cdt_name
- cuda_compiler
- cuda_compiler_version
- docker_image
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ c_stdlib_version:
cdt_arch:
- aarch64
cdt_name:
- cos7
- conda
channel_sources:
- conda-forge
channel_targets:
Expand Down Expand Up @@ -52,7 +52,6 @@ zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cdt_name
- cuda_compiler
- cuda_compiler_version
- docker_image
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ c_stdlib_version:
cdt_arch:
- aarch64
cdt_name:
- cos7
- conda
channel_sources:
- conda-forge
channel_targets:
Expand Down Expand Up @@ -52,7 +52,6 @@ zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cdt_name
- cuda_compiler
- cuda_compiler_version
- docker_image
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ c_stdlib_version:
cdt_arch:
- aarch64
cdt_name:
- cos7
- conda
channel_sources:
- conda-forge
channel_targets:
Expand Down Expand Up @@ -52,7 +52,6 @@ zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cdt_name
- cuda_compiler
- cuda_compiler_version
- docker_image
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ c_stdlib_version:
cdt_arch:
- aarch64
cdt_name:
- cos7
- conda
channel_sources:
- conda-forge
channel_targets:
Expand Down Expand Up @@ -52,7 +52,6 @@ zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cdt_name
- cuda_compiler
- cuda_compiler_version
- docker_image
Expand Down
8 changes: 0 additions & 8 deletions .ci_support/migrations/cudnn9.yaml

This file was deleted.

This file was deleted.

38 changes: 0 additions & 38 deletions .ci_support/migrations/python312.yaml

This file was deleted.

4 changes: 2 additions & 2 deletions .ci_support/osx_64_numpy2.0python3.10.____cpython.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ MACOSX_SDK_VERSION:
c_compiler:
- clang
c_compiler_version:
- '17'
- '18'
c_stdlib:
- macosx_deployment_target
c_stdlib_version:
Expand All @@ -23,7 +23,7 @@ cudnn:
cxx_compiler:
- clangxx
cxx_compiler_version:
- '17'
- '18'
github_actions_labels:
- cirun-openstack-cpu-large
libabseil:
Expand Down
4 changes: 2 additions & 2 deletions .ci_support/osx_64_numpy2.0python3.11.____cpython.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ MACOSX_SDK_VERSION:
c_compiler:
- clang
c_compiler_version:
- '17'
- '18'
c_stdlib:
- macosx_deployment_target
c_stdlib_version:
Expand All @@ -23,7 +23,7 @@ cudnn:
cxx_compiler:
- clangxx
cxx_compiler_version:
- '17'
- '18'
github_actions_labels:
- cirun-openstack-cpu-large
libabseil:
Expand Down
4 changes: 2 additions & 2 deletions .ci_support/osx_64_numpy2.0python3.12.____cpython.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ MACOSX_SDK_VERSION:
c_compiler:
- clang
c_compiler_version:
- '17'
- '18'
c_stdlib:
- macosx_deployment_target
c_stdlib_version:
Expand All @@ -23,7 +23,7 @@ cudnn:
cxx_compiler:
- clangxx
cxx_compiler_version:
- '17'
- '18'
github_actions_labels:
- cirun-openstack-cpu-large
libabseil:
Expand Down
4 changes: 2 additions & 2 deletions .ci_support/osx_64_numpy2python3.13.____cp313.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ MACOSX_SDK_VERSION:
c_compiler:
- clang
c_compiler_version:
- '17'
- '18'
c_stdlib:
- macosx_deployment_target
c_stdlib_version:
Expand All @@ -23,7 +23,7 @@ cudnn:
cxx_compiler:
- clangxx
cxx_compiler_version:
- '17'
- '18'
github_actions_labels:
- cirun-openstack-cpu-large
libabseil:
Expand Down
4 changes: 2 additions & 2 deletions .ci_support/osx_arm64_numpy2.0python3.10.____cpython.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ MACOSX_SDK_VERSION:
c_compiler:
- clang
c_compiler_version:
- '17'
- '18'
c_stdlib:
- macosx_deployment_target
c_stdlib_version:
Expand All @@ -23,7 +23,7 @@ cudnn:
cxx_compiler:
- clangxx
cxx_compiler_version:
- '17'
- '18'
github_actions_labels:
- cirun-openstack-cpu-large
libabseil:
Expand Down
4 changes: 2 additions & 2 deletions .ci_support/osx_arm64_numpy2.0python3.11.____cpython.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ MACOSX_SDK_VERSION:
c_compiler:
- clang
c_compiler_version:
- '17'
- '18'
c_stdlib:
- macosx_deployment_target
c_stdlib_version:
Expand All @@ -23,7 +23,7 @@ cudnn:
cxx_compiler:
- clangxx
cxx_compiler_version:
- '17'
- '18'
github_actions_labels:
- cirun-openstack-cpu-large
libabseil:
Expand Down
4 changes: 2 additions & 2 deletions .ci_support/osx_arm64_numpy2.0python3.12.____cpython.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ MACOSX_SDK_VERSION:
c_compiler:
- clang
c_compiler_version:
- '17'
- '18'
c_stdlib:
- macosx_deployment_target
c_stdlib_version:
Expand All @@ -23,7 +23,7 @@ cudnn:
cxx_compiler:
- clangxx
cxx_compiler_version:
- '17'
- '18'
github_actions_labels:
- cirun-openstack-cpu-large
libabseil:
Expand Down
Loading
Loading