Mathew Odden
ec4b8ee1ed
Fixes for 0.5.0 build ported to rocm-main
...
(cherry picked from commit c23a81461192a2b6da3d364076a261714d2dc64f)
2025-03-25 17:51:30 -05:00
charleshofer
dd7f96b27c
Fix ROCm build README ( #284 )
2025-03-18 14:35:36 -05:00
Mathew Odden
d864b4fbf4
Fix auditwheel version issue ( #288 )
...
Auditwheel 6.3.0 changed/removed the lddtree function
so cap constraint to 6.2.x
2025-03-17 12:30:56 -05:00
charleshofer
f14a1d0b71
Add JSON output to multi-GPU tests ( #274 )
2025-03-12 11:30:55 -05:00
JD
ce53e374fc
Deprecate obsolete gfx versions ( #273 )
2025-03-11 12:19:51 -05:00
JD
a701022ec4
add gfx1101 target ( #249 )
2025-03-03 09:29:47 -06:00
Zahid Iqbal
15255dd69e
removing csv result compilation after Unit test... ( #248 )
2025-03-02 08:30:34 -06:00
Charles Hofer
1217ba9054
Fix numactl-devel name
2025-02-25 20:48:15 +00:00
Charles Hofer
72ecacd870
Fix numa package
2025-02-25 20:41:23 +00:00
Charles Hofer
e82b4e22dc
Install numa library
2025-02-25 20:24:44 +00:00
charleshofer
dd3f34ca2b
Use bazel for PR tests ( #216 )
...
* Use bazel for running pre-merge CI tests
* Don't use HEREDOC
* Fix block text
* Use bash array
* Add bazel install
* Put Bazel in the build image
* Use Bazelisk
* Remove bazel install in Docker
* Go back to upstream XLA
* Remove bazel test command from workflow
* Move test command to build container
* Fix string format typos
2025-02-24 11:38:06 -06:00
Charles Hofer
9133253c20
Change to make CI run
2025-02-11 16:54:51 +00:00
Charles Hofer
4e403d29d2
Remove conflicting param for ci_build
2025-02-10 18:16:21 +00:00
Charles Hofer
31c1f25425
Merge branch 'rocm-main' into ci-upstream-sync-110_1
2025-02-10 18:02:25 +00:00
jax authors
1ca8807dca
Merge pull request #25810 from ROCm:gh-9948-add-gpu-ci-upstream
...
PiperOrigin-RevId: 724378710
2025-02-07 09:51:59 -08:00
charleshofer
ebf4a54f4f
Add AMD ROCm GPU CI post-build check ( #137 )
2025-02-06 21:41:53 +00:00
Charles Hofer
63e6442bdf
Merge branch 'rocm-main' into ci-upstream-sync-97_1
2025-01-27 17:19:08 +00:00
jax authors
9a60e6fce4
Merge pull request #25917 from ROCm:ci_fix_multi_gpu_test_logic-upstream
...
PiperOrigin-RevId: 716153760
2025-01-16 02:45:54 -08:00
Ruturaj4
8e88adcd3f
Fix run_multi_gpu script multi-gpu issue and refactor code
2025-01-15 22:33:03 +00:00
Ruturaj4
435edf1f8c
Add gfx12xx archs
2025-01-15 16:14:40 +00:00
Charles Hofer
1fa6e91af6
Add option to ci_build to run different tests
2025-01-09 16:53:30 +00:00
charleshofer
bc06c93d23
Add GPU CI ( #137 )
2025-01-07 15:01:50 -06:00
Charles Hofer
a94ee1fcb9
Unskip unit tests that are now fixed
2025-01-07 16:56:35 +00:00
charleshofer
4b11080f18
Merge pull request #193 from ROCm/ci-upstream-sync-78_1
...
CI: 01/06/25 upstream sync
2025-01-06 11:22:23 -06:00
Charles Hofer
a1734fd31f
Change to trigger CI
2025-01-06 15:50:02 +00:00
Ruturaj Vaidya
e0ea173e9c
Add upload wheels file for pypi ( #184 )
2025-01-02 11:23:27 -06:00
Ruturaj4
20b75ab82f
Update package indentation fix
2025-01-01 18:50:47 -06:00
jax authors
bad6f0f0e9
Merge pull request #25440 from ROCm:jax_pypi-upstream
...
PiperOrigin-RevId: 707934347
2024-12-19 09:17:46 -08:00
Ruturaj4
fefebebea4
Update documentation and add setup.py pypi bindings
2024-12-19 10:21:28 -06:00
Ruturaj4
bfcace4933
[ROCm] ci build and dockerfile changes
2024-12-19 09:02:55 -06:00
jax authors
7b32d88247
Merge pull request #25136 from ROCm:ci_dockerfile_arg_changes-upstream
...
PiperOrigin-RevId: 701959495
2024-12-02 07:21:15 -08:00
jax authors
ab79066bbe
Merge pull request #25128 from ROCm:ci_fix_wheelhouse_relative_paths-upstream
...
PiperOrigin-RevId: 701143534
2024-11-28 18:51:49 -08:00
jax authors
b0df405250
Merge pull request #25130 from ROCm:ci_fix_set_options-upstream
...
PiperOrigin-RevId: 701143242
2024-11-28 18:49:47 -08:00
jax authors
385e2f4339
Merge pull request #25137 from ROCm:ci_enable_https-upstream
...
PiperOrigin-RevId: 701142951
2024-11-28 18:45:52 -08:00
Ruturaj4
3d8063209e
Update http to https in amd artifactory url.
2024-11-26 22:38:04 +00:00
Ruturaj4
8df2766466
Add argument to override base docker in dockerfile
2024-11-26 22:28:53 +00:00
Ruturaj4
694de6b64c
[ROCm] Change run_multi_gpu set opts
2024-11-26 21:10:37 +00:00
Ruturaj4
d30ec2b5b3
[ROCm] fix jax and wheelhouse relative paths
2024-11-26 21:04:36 +00:00
Nitin Srinivasan
84dc9bab33
Update ROCm scripts to match new build.py usage
2024-11-25 19:25:08 +00:00
Ruturaj4
bfd7075c39
[ROCm] ci build fixes
2024-10-25 05:01:44 -05: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
