ROCm/jax

Pull Requests Commits

[ROCm] wheel install fix

Ruturaj4 committed 1 year ago

eab0bac8

Merge pull request #111 from ROCm/pramenku-patch-2

Ruturaj4 committed 1 year ago

Verified 4083c70b

Correcting docker tag

pramenku committed 1 year ago

Verified 609d35d5

[ROCm] Fix invalid repo url for EL path

mrodden committed 1 year ago

d0829370

[JAX] fix jax and wheelhouse relative paths

Ruturaj4 committed 1 year ago

eca7df07

Fix formatting issues

mrodden committed 1 year ago

95115832

Fix DOCKER_BUILDKIT issue for jax CI

kiran-thumma committed 1 year ago

Verified 2a0e2125

Add docker builds for ubu22 and 24

mrodden committed 1 year ago

18dd05a8

Fix typo in run_single_gpu

mrodden committed 1 year ago

658d84cb

[ROCM] make mosaic dependency cuda specific

Ruturaj4 committed 1 year ago

88c89ca6

* Add conditional docker interactive mode

mrodden committed 1 year ago

eb78dfd1

[ROCm] improve gpu script

Ruturaj4 committed 1 year ago

58c43201

[ROCm] Fix python in rocm ci_build script.

Ruturaj4 committed 1 year ago

dbf6b1e4

Add missing CPython build deps for pyenv

mrodden committed 1 year ago

cc2632b1

Fix not finding wheels in bazel output

mrodden committed 1 year ago

3084b0e5

Fix jenkins notty issue

mrodden committed 1 year ago

67b3a49d

Add defaults for ROCm build vars

mrodden committed 1 year ago

92583b5b

Add copyright and license headers to new files

mrodden committed 1 year ago

f729724c

Add support for ROCm development builds

mrodden committed 1 year ago

2d39de54

Add internal release support to get_rocm.py

mrodden committed 1 year ago

d3cd81a8

[ROCm] Change ROCm builds to manylinux wheels

mrodden committed 1 year ago

effbde74

Fix lint in run_single_gpu.py

yueshengys committed 1 year ago

120d1158

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

Rahul Batra committed 1 year ago

787406a5

Fix pre-commit formatting errors

dfm committed 1 year ago

71273018

Update build.py

Ruturaj4 committed 1 year ago

b87fc1be

[ROCM] nits and fixes

Ruturaj4 committed 1 year ago

4523f60c

[JAX] change rocm build file for pjrt plugin

Ruturaj4 committed 1 year ago

ac09998b

[ROCm] Bring-up pjrt support

Ruturaj4 committed 1 year ago

33d88618

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

Ruturaj4 committed 1 year ago

d197a6e8

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

kiran-thumma committed 1 year ago

Verified b3d1a641

Older