mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-22 20:36:04 +00:00
Code Issues Packages Projects Releases Wiki Activity
23,755 Commits 508 Branches 263 Tags
Commit Graph

86 Commits

Author SHA1 Message Date
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
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
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
Mathew Odden
a9e54b3e0a Add docker builds for ubu22 and 24 2024-08-27 16:40:37 -05:00
jax authors
140955dce0 Merge pull request #23224 from ROCm:ci_build_script 
PiperOrigin-RevId: 668061975
 2024-08-27 11:12:05 -07:00
Ayaka
859eacb5a1 Fix mypy error 2024-08-27 16:57:53 +01:00
Ruturaj4
9ce8de5fb0 [ROCm] add build file. 2024-08-23 18:11:48 -05:00
Mathew Odden
5c2ffa893f * Add conditional docker interactive mode 
Interactive causes bazel to output more
useful info when running locally.

* Fix issue with rocm el8 repo urls

Work around quirk with rocm version
when it ends with 0

* Fix package name conflict

Ubu22 and higher have a package name conflict
between the debian versions and the AMD provided
versions.

* [ROCm] Use clang env
 2024-08-22 10:08:41 -05:00
Ruturaj4
e06be544d4 [ROCm] improve gpu script 2024-08-19 15:13:58 -05:00
Ruturaj4
fd7c52d213 [ROCm] Fix python in rocm ci_build script. 2024-08-15 21:33:51 -05:00
Mathew Odden
fafa03c60f Add missing CPython build deps for pyenv 2024-08-12 15:01:34 -05:00
Mathew Odden
701cda8ebd Fix not finding wheels in bazel output 2024-08-12 15:01:34 -05:00
Mathew Odden
df2d140f51 Fix jenkins notty issue 2024-08-12 15:01:34 -05:00
Mathew Odden
319ebf81c1 Add defaults for ROCm build vars 2024-08-12 15:01:34 -05:00