18 Commits

Author SHA1 Message Date
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
charleshofer
ebf4a54f4f Add AMD ROCm GPU CI post-build check (#137) 2025-02-06 21:41:53 +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
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
Ruturaj4
8df2766466 Add argument to override base docker in dockerfile 2024-11-26 22:28:53 +00:00
Ruturaj4
d30ec2b5b3 [ROCm] fix jax and wheelhouse relative paths 2024-11-26 21:04:36 +00:00
Ruturaj4
a2824862f5 [ROCm] build script fix 2024-10-15 13:43:08 -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
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
fd7c52d213 [ROCm] Fix python in rocm ci_build script. 2024-08-15 21:33:51 -05:00
Mathew Odden
df2d140f51 Fix jenkins notty issue 2024-08-12 15:01:34 -05:00
Mathew Odden
abe44f6d9e Add copyright and license headers to new files 2024-08-12 15:01:34 -05:00
Mathew Odden
a1a0a4ecdd Add support for ROCm development builds
Use get_rocm.py changes in ci_build to pull in
development builds for ROCm.

Specify ROCM_BUILD_JOB and ROCM_BUILD_NUM for
activating the development build path.
2024-08-12 15:01:34 -05:00
Mathew Odden
1e58d76772 [ROCm] Change ROCm builds to manylinux wheels 2024-08-12 15:01:34 -05:00