ROCm/jax

Pull Requests Commits

unskipping tests that were previously skipped for HIP runtime issue

root committed 192 days ago

a4783eb4

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

charleshofer committed 196 days ago

Verified db216443

Remove old build scripts (#532)

charleshofer committed 200 days ago

Verified d9b4b407

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

gulsumgudukbay committed 223 days ago

Verified 4ee8db98

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

gulsumgudukbay committed 224 days ago

Verified c07c3a15

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

gulsumgudukbay committed 224 days ago

Verified 68d7228c

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

gulsumgudukbay committed 225 days ago

Verified f697dde0

unskip pallas tests that were skipped due to LLVm error

gulsumgudukbay committed 225 days ago

41a32e35

addressing Charlie's comments

gulsumgudukbay committed 231 days ago

248cf432

removed a print stmt

gulsumgudukbay committed 231 days ago

e1236c74

Adding abort test info in run_single_gpu.py logs

gulsumgudukbay committed 231 days ago

1ff07701

fix is_device_rocm function call

root committed 234 days ago

bcc84983

Fix GPU handle pool singleton aliasing (#525)

phambinhfin committed 238 days ago

Verified 9890a51d

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

Arech8 committed 239 days ago

Verified 2c4c7064

adding additional skips only for MI350 for LLVM cannot scalarize ERROR

root committed 240 days ago

e5fc666b

Bugfix bazel debug rule (#511)

Arech8 committed 241 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 254 days ago

3d6b521d

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

Arech8 committed 259 days ago

Verified 03684216

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

Arech8 committed 260 days ago

58a5c560

Revert "unskip test_select_n"

gulsumgudukbay committed 260 days ago

81ba3269

skip test_fused_attention Interpret mode

gulsumgudukbay committed 261 days ago

5eed81de

unskip test_select_n

gulsumgudukbay committed 263 days ago

8bf16ba1

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

zahiqbal committed 269 days ago

7acefb20

skip more tests

gulsumgudukbay committed 270 days ago

5271d8f5

gulsumgudukbay committed 270 days ago

6d1f1ac1

Update scipy_signal_test.py

gulsumgudukbay committed 270 days ago

e1e0e2ed

skip additional tests

gulsumgudukbay committed 270 days ago

2ce571c9

now builds with --rocm_version=7 flag set instead of 60 only

gulsumgudukbay committed 278 days ago

f12689aa

[ROCM] Add missing dependency on amdhip64 runtime

mrodden committed 278 days ago

d9047090

[ROCm] Add rocm7 plugin namespace to jaxlib ext's

mrodden committed 278 days ago

3a6435f6

Older