ROCm/jax

Pull Requests Commits

[ROCm] Use clang env

Ruturaj4 committed 2 years ago

330c4f6f

[ROCm] Fix invalid repo url for EL path

mrodden committed 2 years ago

d0829370

[JAX] fix jax and wheelhouse relative paths

Ruturaj4 committed 2 years ago

eca7df07

Fix formatting issues

mrodden committed 2 years ago

95115832

Fix DOCKER_BUILDKIT issue for jax CI

kiran-thumma committed 2 years ago

Verified 2a0e2125

Add docker builds for ubu22 and 24

mrodden committed 2 years ago

18dd05a8

Fix typo in run_single_gpu

mrodden committed 2 years ago

658d84cb

[ROCM] make mosaic dependency cuda specific

Ruturaj4 committed 2 years ago

88c89ca6

* Add conditional docker interactive mode

mrodden committed 2 years ago

eb78dfd1

[ROCm] improve gpu script

Ruturaj4 committed 2 years ago

58c43201

[ROCm] Fix python in rocm ci_build script.

Ruturaj4 committed 2 years ago

dbf6b1e4

Add missing CPython build deps for pyenv

mrodden committed 2 years ago

cc2632b1

Fix not finding wheels in bazel output

mrodden committed 2 years ago

3084b0e5

Fix jenkins notty issue

mrodden committed 2 years ago

67b3a49d

Add defaults for ROCm build vars

mrodden committed 2 years ago

92583b5b

Add copyright and license headers to new files

mrodden committed 2 years ago

f729724c

Add support for ROCm development builds

mrodden committed 2 years ago

2d39de54

Add internal release support to get_rocm.py

mrodden committed 2 years ago

d3cd81a8

[ROCm] Change ROCm builds to manylinux wheels

mrodden committed 2 years ago

effbde74

Fix lint in run_single_gpu.py

yueshengys committed 2 years ago

120d1158

[ROCm]: Add support to continue on fail, fix script paths and update Dockerfile to add necessary packages

Rahul Batra committed 2 years ago

787406a5

Fix pre-commit formatting errors

dfm committed 2 years ago

71273018

Update build.py

Ruturaj4 committed 2 years ago

b87fc1be

[ROCM] nits and fixes

Ruturaj4 committed 2 years ago

4523f60c

[JAX] change rocm build file for pjrt plugin

Ruturaj4 committed 2 years ago

ac09998b

[ROCm] Bring-up pjrt support

Ruturaj4 committed 2 years ago

33d88618

[ROCm] Add missing report-log pakcage for testing.

Ruturaj4 committed 2 years ago

d197a6e8

Merge pull request #67 from ROCm/rocm-jaxlib-v0.4.30-kithumma

kiran-thumma committed 2 years ago

Verified b3d1a641

kiran-thumma committed 2 years ago

3545fa34

update docker 0.4.30

kiran-thumma committed 2 years ago

a9c3af7d

Older