srayasam-amd
40a993797a
Update ci_build
2025-02-13 18:46:06 +05:30
srayasam-amd
ae9f449c13
Update ci_build
2025-02-13 17:15:13 +05:30
srayasam-amd
c95d267620
Update build_wheels.py
2025-02-13 16:06:56 +05:30
srayasam-amd
90f95a2e57
Update ci_build
2025-02-13 15:41:25 +05:30
srayasam-amd
2aa8e23b93
Update Dockerfile.manylinux_2_28_x86_64.rocm
2025-02-13 11:17:53 +05:30
srayasam-amd
96315972a8
Update ci_build.sh
2025-02-13 11:06:55 +05:30
srayasam-amd
082d05757f
Update ci_build.sh
2025-02-13 11:01:03 +05:30
srayasam-amd
d3cdd4f72d
Update ci_build.sh
2025-02-13 10:56:45 +05:30
srayasam-amd
6f8b984b89
Update Dockerfile.manylinux_2_28_x86_64.rocm
2025-02-13 10:51:17 +05:30
srayasam-amd
3e64d0b627
Update ci_build.sh
2025-02-13 10:44:44 +05:30
srayasam-amd
6d50f2a2c2
Update ci_build.sh
2025-02-13 10:30:47 +05:30
srayasam-amd
bdea57c455
Update Dockerfile.manylinux_2_28_x86_64.rocm
2025-02-13 10:27:00 +05:30
srayasam-amd
d917a22513
Update Dockerfile.manylinux_2_28_x86_64.rocm
2025-02-12 21:07:02 +05:30
srayasam-amd
3ba06823e8
Update Dockerfile.manylinux_2_28_x86_64.rocm
2025-02-12 15:39:06 +05:30
srayasam-amd
58f2c0e44d
Update ci_build.sh
2025-02-12 14:52:10 +05:30
srayasam-amd
3403473e91
Update ci_build.sh
2025-02-12 13:46:14 +05:30
srayasam-amd
5001365663
Update ci_build.sh
2025-02-12 11:56:00 +05:30
srayasam-amd
a53822fa7c
Update ci_build.sh
...
adding if condition for gpu_Device_targets
2025-02-12 11:53:57 +05:30
srayasam-amd
5da1c4e8df
Update ci_build
2025-02-10 18:08:40 +05:30
srayasam-amd
460b3c357d
Update build_wheels.py
2025-02-10 18:02:52 +05:30
srayasam-amd
f72eb95f6d
Update ci_build.sh
2025-02-10 18:01:30 +05:30
srayasam-amd
a3d3481867
Update Dockerfile.manylinux_2_28_x86_64.rocm
...
adding gpu_device_targets
2025-02-10 17:56:51 +05:30
Ruturaj Vaidya
da40f36d00
Fix run_multi_gpu script multi-gpu issue and refactor code ( #208 )
2025-01-19 18:03:00 -06:00
Ruturaj Vaidya
f99fbcb3c5
Add gfx12xx archs ( #205 )
2025-01-14 10:52:00 -06:00
kithumma
f075f904b2
buildkit option for docker build
2025-01-11 06:09:27 +00:00
kithumma
590ac67c4b
fix cdn
2024-12-30 17:11:51 +00:00
Ruturaj Vaidya
20362f83c6
[ROCm] Remove exit on error ( #167 )
2024-12-02 06:51:25 -06:00
Ruturaj4
010aea0c56
Add argument to override base docker in dockerfile
2024-12-02 06:49:40 -06:00
Ruturaj4
3b3083bbc4
[ROCm] ci build and dockerfile changes
2024-12-02 06:49:32 -06:00
Ruturaj4
b8d2dedeed
[ROCm] Change run_multi_gpu set opts
2024-12-02 06:49:22 -06:00
Ruturaj4
a5a1417587
BUILD file fixes and linter run
2024-12-02 06:49:15 -06:00
Ruturaj4
addca5d517
[ROCm] fix jax and wheelhouse relative paths
2024-12-02 06:49:10 -06:00
Ruturaj4
39c0450638
[ROCm] ci build fixes
2024-12-02 06:49:01 -06:00
jax authors
3bdc57dd29
Merge pull request #24300 from ROCm:ci_rocm_readme
...
PiperOrigin-RevId: 686872994
2024-10-17 05:21:13 -07:00
Ruturaj4
3c3b08dfd6
[ROCm] Fix README.md to update AMD JAX installation instructions
2024-10-16 17:15:32 -05:00
Ruturaj4
a2824862f5
[ROCm] build script fix
2024-10-15 13:43:08 -05:00
Ruturaj4
937d79e3f2
[ROCm] apt update
2024-10-11 18:33:12 -05:00
jax authors
e4629f6a4c
Merge pull request #24232 from ROCm:ci_rv_clang_clean
...
PiperOrigin-RevId: 684891301
2024-10-11 11:00:55 -07:00
David Dunleavy
ee312afe86
Corresponding build.py updates after 5132a188f7
...
PiperOrigin-RevId: 684805296
2024-10-11 05:41:28 -07:00
Ruturaj4
33bcd0cb7a
[ROCm] Bring up clang support for JAX+XLA
...
* Add clang path
* bazelrc env fixes
* Fix wheelhouse installation and preserve wheels
* dockerfile changes
* Add target.lst
* Change target architectures
* Install bzip2 and sqlite packages
2024-10-10 16:31:26 -05:00
jax authors
53668b88eb
Update rules_python.patch to support Python 3.13.0 and update python 3.13 packages in JAX.
...
The downloaded Python `tar.gz` files should have suffix `install_only`.
PiperOrigin-RevId: 684113192
2024-10-09 11:36:33 -07:00
jax authors
e212c77336
Merge pull request #23891 from ROCm:build-fixes-rollup
...
PiperOrigin-RevId: 681448694
2024-10-02 07:43:13 -07:00
Mathew Odden
9ff891dfa1
[ROCm] Remove broken legacy env vars
...
These env vars are no longer used or need and were
being set incorrectly.
[ROCm] Use specific amdgpu version for EL8 systems
We were always installing the latest driver versions
but this had some side effects when yum would try
to download index files from a URL with changing content.
[ROCm] Fix formatting on python files
Reformatted with black
2024-09-30 12:39:51 -05:00
jax authors
f1b3251bf9
Change CLANG_CUDA_COMPILER_PATH set order. Add --config=cuda_clang to build.py
...
Set `--action_env=CLANG_CUDA_COMPILER_PATH` after cuda_nvcc configuration
Add `--config=cuda_clang` when `--nouse_cuda_nvcc` flag set
PiperOrigin-RevId: 678873849
2024-09-25 15:39:44 -07:00
8bitmp3
60a06fd4c9
Update pillow version in JAX build test-requirements.txt
2024-09-25 14:55:46 +00:00
jax authors
6e116491c1
Add --use_cuda_nvcc flag to enable or disable compilation of CUDA code using NVCC.
...
If `--use_cuda_nvcc` flag is set the NVCC compiler driver will be used to build the CUDA code (default behavior). Otherwise, if the flag `--nouse_cuda_nvcc` is set, only the clang compiler will be used to build the CUDA code (effectively disabling NVCC).
Mark `--use_clang` flag as deprecated.
Refactor `.bazelrc` configs to match the new flag and to cleanup all previous confusing names.
PiperOrigin-RevId: 678332548
2024-09-24 11:37:00 -07:00
jax authors
0c7c71e640
Update python version from 3.12 to 3.13.0rc2 in Github presubmit jobs.
...
PiperOrigin-RevId: 676140293
2024-09-18 14:49:42 -07:00
jax authors
8bcdb12852
Add CI jobs for python 3.13.0rc2.
...
PiperOrigin-RevId: 675758096
2024-09-17 16:51:35 -07:00
Vadym Matsishevskyi
8804be0229
Add Python 3.130rc2 support to the build.
...
This PR depends on https://github.com/openxla/xla/pull/17169 . The change does not fail existing builds, but to be able to use python 3.13 functionality in jax the corresponding XLA pr needs to land first and get integrated with JAX (happens automatically).
PiperOrigin-RevId: 675243989
2024-09-16 12:14:32 -07:00
Mathew Odden
a9e54b3e0a
Add docker builds for ubu22 and 24
2024-08-27 16:40:37 -05:00
