ROCm/jax

Pull Requests Commits

guard -it for docker with check

JehandadKhan committed 1 year ago

6a0d1a63

ensure stdin is a tty

JehandadKhan committed 1 year ago

8a6a2c93

JehandadKhan committed 1 year ago

23dcab16

override image name

JehandadKhan committed 1 year ago

4df858b7

fix typo in script

JehandadKhan committed 1 year ago

f0125ad9

update permissions for wheels

JehandadKhan committed 1 year ago

47521d09

fix action version

JehandadKhan committed 1 year ago

b7a98485

JehandadKhan committed 1 year ago

8de5fada

remove cleanup stage

JehandadKhan committed 1 year ago

6a49a56d

JehandadKhan committed 1 year ago

e9689d54

remove the egg-info dir

JehandadKhan committed 1 year ago

c62d4b40

delete the dist directory

JehandadKhan committed 1 year ago

5a50530d

JehandadKhan committed 1 year ago

0bf3f5de

JehandadKhan committed 1 year ago

c8c858f0

fix run id typo

JehandadKhan committed 1 year ago

c5131001

update amdgpu repo url to include version number

JehandadKhan committed 1 year ago

2928bb6b

fix the pwd issue in docker, string fails to decode

JehandadKhan committed 1 year ago

74e60252

remove redundant step

JehandadKhan committed 1 year ago

89dfac4b

Build doker in workflow

JehandadKhan committed 1 year ago

d41c479b

Merge branch 'main' into jd/gpu_ci

JehandadKhan committed 1 year ago

f8574788

Include the sdy MLIR dialect in jaxlib.

hawkinsp committed 1 year ago

5a1d0a6c

Replace disable_backends with enable_backends on jax_multiplatform_test.

hawkinsp committed 1 year ago

26632fd3

[Pallas/MGPU] Skip output transfers when they don't depend on sequenital dims

apaszke committed 1 year ago

5740ab3b

Run Pallas Mosaic GPU tests on internal CI

superbobry committed 1 year ago

afaf8b82

Fixed Pallas Mosaic GPU test following recent changes

superbobry committed 1 year ago

3ae48621

[Pallas:TPU] Fix lowering of convert_element_type(int32) -> bool.

Google-ML-Automation committed 1 year ago

ea86251a

Removed unused imports in `jax.experimental.mosaic.gpu.core`

superbobry committed 1 year ago

ea6ee4d7

[Pallas TPU] Add a note in the Pallas Quickstart documentation about the instructions of running the existing example on TPU

ayaka14732 committed 1 year ago

ab4590ce

Merge pull request #23853 from zhenying-liu:remat-scan

Google-ML-Automation committed 1 year ago

5a1549cc

[XLA:Mosaic][Pallas] Enable vector.ExtractOp for non-zero indices.

justinjfu committed 1 year ago

9f4e8d00

Older