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.

Sign up for GitHub

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 and add fixes for CUDA #288

Merged
merged 11 commits into from
Dec 9, 2024
+246 −449
Merged

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 #288

Show file tree
Hide file tree
Changes from 7 commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
e415cb7
Generate also jax-cuda-plugin and jax-cuda-pjrt in cuda builds
traversaro Nov 18, 2024
501323c
Bump build number
traversaro Nov 18, 2024
1030732
MNT: Re-rendered with conda-build 24.9.0, conda-smithy 3.44.6, and co…
Nov 18, 2024
9a79392
Copy all cudnn headers
traversaro Nov 19, 2024
0b8cabf
Fix linking errors
traversaro Nov 19, 2024
43a9420
Fix abseil log init
xhochy Nov 22, 2024
013cc28
MNT: Re-rendered with conda-build 24.9.0, conda-smithy 3.44.7, and co…
xhochy Nov 22, 2024
692c3fa
Add -labsl_log_internal_globals
xhochy Nov 22, 2024
266fbcd
MNT: Re-rendered with conda-build 24.9.0, conda-smithy 3.44.7, and co…
h-vetinari Nov 24, 2024
8c12eaf
Patch xla to find conda cuda libraries and add dependency on required…
traversaro Dec 8, 2024
b4b6b5b
Trigger CI
xhochy Dec 9, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions ...da-nvcccuda_compiler_version12.0cxx_compiler_version12numpy2.0python3.10.____cpython.yaml
View file
Open in desktop
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
3 changes: 1 addition & 2 deletions ...da-nvcccuda_compiler_version12.0cxx_compiler_version12numpy2.0python3.11.____cpython.yaml
View file
Open in desktop
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
3 changes: 1 addition & 2 deletions ...da-nvcccuda_compiler_version12.0cxx_compiler_version12numpy2.0python3.12.____cpython.yaml
View file
Open in desktop
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
3 changes: 1 addition & 2 deletions ...ercuda-nvcccuda_compiler_version12.0cxx_compiler_version12numpy2python3.13.____cp313.yaml
View file
Open in desktop
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
3 changes: 1 addition & 2 deletions ...lerNonecuda_compiler_versionNonecxx_compiler_version13numpy2.0python3.10.____cpython.yaml
View file
Open in desktop
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
3 changes: 1 addition & 2 deletions ...lerNonecuda_compiler_versionNonecxx_compiler_version13numpy2.0python3.11.____cpython.yaml
View file
Open in desktop
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
3 changes: 1 addition & 2 deletions ...lerNonecuda_compiler_versionNonecxx_compiler_version13numpy2.0python3.12.____cpython.yaml
View file
Open in desktop
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
3 changes: 1 addition & 2 deletions ...ompilerNonecuda_compiler_versionNonecxx_compiler_version13numpy2python3.13.____cp313.yaml
View file
Open in desktop
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
3 changes: 1 addition & 2 deletions ...lerNonecuda_compiler_versionNonecxx_compiler_version13numpy2.0python3.10.____cpython.yaml
View file
Open in desktop
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
3 changes: 1 addition & 2 deletions ...lerNonecuda_compiler_versionNonecxx_compiler_version13numpy2.0python3.11.____cpython.yaml
View file
Open in desktop
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
3 changes: 1 addition & 2 deletions ...lerNonecuda_compiler_versionNonecxx_compiler_version13numpy2.0python3.12.____cpython.yaml
View file
Open in desktop
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
3 changes: 1 addition & 2 deletions ...ompilerNonecuda_compiler_versionNonecxx_compiler_version13numpy2python3.13.____cp313.yaml
View file
Open in desktop
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
View file
Open in desktop

This file was deleted.

18 changes: 0 additions & 18 deletions .ci_support/migrations/libabseil20240722_grpc165_libprotobuf5275.yaml
View file
Open in desktop

This file was deleted.

38 changes: 0 additions & 38 deletions .ci_support/migrations/python312.yaml
View file
Open in desktop

This file was deleted.

7 changes: 1 addition & 6 deletions .github/workflows/conda-build.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -137,12 +137,6 @@ jobs:
fi
./.scripts/run_osx_build.sh

- name: Install Miniconda for windows
uses: conda-incubator/setup-miniconda@a4260408e20b96e80095f42ff7f1a15b27dd94ca # v3.0.4
with:
miniforge-version: latest
if: matrix.os == 'windows'

- name: Build on windows
shell: cmd
run: |
Expand All @@ -151,6 +145,7 @@ jobs:
set "sha=%GITHUB_SHA%"
call ".scripts\run_win_build.bat"
env:
MINIFORGE_HOME: D:\Miniforge
PYTHONUNBUFFERED: 1
CONFIG: ${{ matrix.CONFIG }}
CI: github_actions
Expand Down
10 changes: 5 additions & 5 deletions .scripts/build_steps.sh
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

41 changes: 26 additions & 15 deletions .scripts/run_osx_build.sh
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion azure-pipelines.yml
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

7 changes: 7 additions & 0 deletions build-locally.py
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading
Loading