ROCm/jax

Pull Requests Commits

[Fix] breaking BUILD file with rocm_hip rename (#548)

GeneDer committed 107 days ago

Verified ae78eb36

Updating jaxlib BUILD file for new target and XLA commit hash (#546)

gulsumgudukbay committed 107 days ago

Verified 9a2d48ea

Wrap HIP inline functions in anonymous namespaces in vendor.h (#545)

AratiGanesh committed 110 days ago

Verified b5397a5e

[0.6.0-UT] Adding third field to return value of get_rocm_version (#542)

gulsumgudukbay committed 124 days ago

Verified 9fc59609

[0.6.0-UT] Adding python equivalent of run_multi_gpu.sh (#543)

gulsumgudukbay committed 125 days ago

Verified 4528432b

[0.6.0-UT] Unskipping passing tests (#540)

gulsumgudukbay committed 125 days ago

Verified 4b3d54ab

Fix GPU handle pool singleton aliasing for HipSPARSE (#535)

AratiGanesh committed 126 days ago

Verified f1aaf9a9

Revert "[0.6.0-UT] Skipping aborted tests (HIP runtime issue)" (#539)

gulsumgudukbay committed 127 days ago

Verified 1a647fa6

Revert "Remove old build scripts (#532)" (#536)

charleshofer committed 131 days ago

Verified db216443

Remove old build scripts (#532)

charleshofer committed 135 days ago

Verified d9b4b407

[0.6.0-UT] skipped more tests for build 33 (#531)

gulsumgudukbay committed 158 days ago

Verified 4ee8db98

[0.6.0-UT] Skipping aborted tests (HIP runtime issue) (#530)

gulsumgudukbay committed 159 days ago

Verified c07c3a15

[0.6.0-UT] Adding abort support to run_multi_gpu (#529)

gulsumgudukbay committed 159 days ago

Verified 68d7228c

[0.6.0-UT] Transferring Multi-GPU tests (#527)

gulsumgudukbay committed 159 days ago

Verified f697dde0

unskip pallas tests that were skipped due to LLVm error

gulsumgudukbay committed 160 days ago

41a32e35

addressing Charlie's comments

gulsumgudukbay committed 166 days ago

248cf432

removed a print stmt

gulsumgudukbay committed 166 days ago

e1236c74

Adding abort test info in run_single_gpu.py logs

gulsumgudukbay committed 166 days ago

1ff07701

fix is_device_rocm function call

root committed 169 days ago

bcc84983

Fix GPU handle pool singleton aliasing (#525)

phambinhfin committed 173 days ago

Verified 9890a51d

Skip test_dot_precision() with TF32 for Navi architecture (#520)

Arech8 committed 174 days ago

Verified 2c4c7064

adding additional skips only for MI350 for LLVM cannot scalarize ERROR

root committed 175 days ago

e5fc666b

Bugfix bazel debug rule (#511)

Arech8 committed 176 days ago

Verified 9c088fc9

The reshape call has a numerical bug where the target reshape size is always higher than array size if num_devices is not a nonnegative power of two. To fix this, this commit adds padding

gulsumgudukbay committed 189 days ago

3d6b521d

Fix test_gpu_distributed_initialize() to get a more proper GPUs number (#503)

Arech8 committed 194 days ago

Verified 03684216

Fix test_gpu_distributed_initialize() to take number of available GPUs into account

Arech8 committed 195 days ago

58a5c560

Revert "unskip test_select_n"

gulsumgudukbay committed 195 days ago

81ba3269

skip test_fused_attention Interpret mode

gulsumgudukbay committed 196 days ago

5eed81de

unskip test_select_n

gulsumgudukbay committed 197 days ago

8bf16ba1

Adjusting block sizes in BWD-FusedAttention tests to fit in GPU shared memory

zahiqbal committed 204 days ago

7acefb20

Older