ROCm/jax

Pull Requests Commits

updating jaxlib BUILD file for new target along with XLA commit for build

gulsumgudukbay committed 75 days ago

8e67a169

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

AratiGanesh committed 78 days ago

Verified b5397a5e

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

gulsumgudukbay committed 92 days ago

Verified 9fc59609

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

gulsumgudukbay committed 93 days ago

Verified 4528432b

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

gulsumgudukbay committed 93 days ago

Verified 4b3d54ab

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

AratiGanesh committed 94 days ago

Verified f1aaf9a9

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

gulsumgudukbay committed 95 days ago

Verified 1a647fa6

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

charleshofer committed 99 days ago

Verified db216443

Remove old build scripts (#532)

charleshofer committed 103 days ago

Verified d9b4b407

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

gulsumgudukbay committed 126 days ago

Verified 4ee8db98

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

gulsumgudukbay committed 127 days ago

Verified c07c3a15

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

gulsumgudukbay committed 127 days ago

Verified 68d7228c

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

gulsumgudukbay committed 127 days ago

Verified f697dde0

unskip pallas tests that were skipped due to LLVm error

gulsumgudukbay committed 128 days ago

41a32e35

addressing Charlie's comments

gulsumgudukbay committed 134 days ago

248cf432

removed a print stmt

gulsumgudukbay committed 134 days ago

e1236c74

Adding abort test info in run_single_gpu.py logs

gulsumgudukbay committed 134 days ago

1ff07701

fix is_device_rocm function call

root committed 137 days ago

bcc84983

Fix GPU handle pool singleton aliasing (#525)

phambinhfin committed 141 days ago

Verified 9890a51d

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

Arech8 committed 142 days ago

Verified 2c4c7064

adding additional skips only for MI350 for LLVM cannot scalarize ERROR

root committed 143 days ago

e5fc666b

Bugfix bazel debug rule (#511)

Arech8 committed 144 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 157 days ago

3d6b521d

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

Arech8 committed 162 days ago

Verified 03684216

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

Arech8 committed 163 days ago

58a5c560

Revert "unskip test_select_n"

gulsumgudukbay committed 163 days ago

81ba3269

skip test_fused_attention Interpret mode

gulsumgudukbay committed 164 days ago

5eed81de

unskip test_select_n

gulsumgudukbay committed 165 days ago

8bf16ba1

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

zahiqbal committed 172 days ago

7acefb20

skip more tests

gulsumgudukbay committed 173 days ago

5271d8f5

Older