ROCm/jax

Pull Requests Commits

Add clang installation to manylinux docker image

Ruturaj4 committed 1 year ago

5d83339d

Update http to https as http is disabled (#144)

pramenku committed 1 year ago

Verified 58e9a215

Make base docker as ARG to override (#134)

pramenku committed 1 year ago

Verified 7d0d2dd1

fix docker exit case (#127)

pramenku committed 1 year ago

Verified c685be5b

Merge pull request #122 from ROCm/ci_clang_31

pramenku committed 1 year ago

Verified 52a31761

Ruturaj4 committed 1 year ago

f4f860e6

Install gcc-c++ and remove clang installation

Ruturaj4 committed 1 year ago

d2f26e3d

Change run_multi_gpu set opts

mrodden committed 1 year ago

023b8c67

BUILD file fixes and linter run

Ruturaj4 committed 1 year ago

584e806e

[ROCm] Use specific amdgpu version for EL8 systems

mrodden committed 1 year ago

9388c17b

[ROCm] Bring up clang support for JAX+XLA

Ruturaj4 committed 1 year ago

952f5635

Update Dockerfile.ms

Ruturaj4 committed 1 year ago

fe96552e

[ROCm] apt update

Ruturaj4 committed 1 year ago

8189eb8c

pramenku committed 1 year ago

Verified ba8d5add

[ROCm]: Add get_arch_details for triton kernel call

Rahul Batra committed 1 year ago

8c1e6b9f

[ROCm] wheel install fix

Ruturaj4 committed 1 year ago

5fb90df1

Correcting docker tag

pramenku committed 1 year ago

d33807c1

[JAX] fix jax and wheelhouse relative paths

Ruturaj4 committed 1 year ago

a92f7f05

[ROCm] Fix invalid repo url for EL path

mrodden committed 1 year ago

de61e3fc

Fix formatting issues

mrodden committed 1 year ago

96a25861

Fix DOCKER_BUILDKIT issue for jax CI

kiran-thumma committed 1 year ago

70ee4c23

Add docker builds for ubu22 and 24

mrodden committed 1 year ago

a2ab4acf

ayaka14732 committed 1 year ago

f02f2a7c

[ROCm] add build file.

Ruturaj4 committed 1 year ago

9b9ea337

* Add conditional docker interactive mode

mrodden committed 1 year ago

a26aa082

[ROCm] improve gpu script

Ruturaj4 committed 1 year ago

3192b8ac

[ROCm] Fix python in rocm ci_build script.

Ruturaj4 committed 1 year ago

1d45b98e

Add missing CPython build deps for pyenv

mrodden committed 1 year ago

f0a6a8b3

Fix not finding wheels in bazel output

mrodden committed 1 year ago

74401c33

Fix jenkins notty issue

mrodden committed 1 year ago

f4e1188b

Older