Ruturaj4
8df2766466
Add argument to override base docker in dockerfile
2024-11-26 22:28:53 +00:00
Ruturaj4
bfd7075c39
[ROCm] ci build fixes
2024-10-25 05:01:44 -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
Mathew Odden
319ebf81c1
Add defaults for ROCm build vars
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
Rahul Batra
b7a7f0bd80
[ROCm]: Dockerfile updates
2024-01-22 16:08:37 +00:00
Rahul Batra
ef79c19093
[ROCm]: Dockerfile and build script updates
...
Add hipblaslt in Dockerfile
Update docker file to default to ROCm5.6
CI scripts update to handle multiple ROCm versions
2023-09-08 22:56:59 +00:00
Rahul Batra
b0e541a730
[ROCm]: Updates for container and build script
...
-Updated dockerfile.ms
-Updated build script to switch building against XLA repo
-Update CI script
-Update jaxlib setup.py to add rocm version
2023-06-19 18:13:28 +00:00
Chao Chen
a2c9fc02e4
jax-rocm runtime/ci dockerfile multistages
2022-12-12 07:45:12 -08:00
Jason Furmanek
34f6646050
Add default setting for TENSORFLOW_ROCM_COMMIT
2022-10-07 19:57:53 +00:00
Jason Furmanek
0783f8982d
[ROCM] Add TENSORFLOW_ROCM_COMMIT parameter to ROCM ci build
2022-09-29 12:40:40 +00:00
Peter Hawkins
ba557d5e1b
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
...
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.
PiperOrigin-RevId: 476167538
2022-09-22 12:27:19 -07:00
jax authors
cf9a900d78
Merge pull request #9584 from ROCmSoftwarePlatform:rocm_refactor_jaxlib
...
PiperOrigin-RevId: 432236852
2022-03-03 11:11:02 -08:00
Reza Rahimi
a0d9d81f92
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
