22 Commits

Author SHA1 Message Date
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
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
Ruturaj4
435edf1f8c Add gfx12xx archs 2025-01-15 16:14:40 +00: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
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
Nitin Srinivasan
84dc9bab33 Update ROCm scripts to match new build.py usage 2024-11-25 19:25:08 +00: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
Ayaka
859eacb5a1 Fix mypy error 2024-08-27 16:57:53 +01: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
701cda8ebd Fix not finding wheels in bazel output 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
3175f13c59 Add internal release support to get_rocm.py 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