ROCm/jax

Pull Requests Commits

unskipping tests that were previously skipped for HIP runtime issue

root committed 361 days ago

a4783eb4

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

charleshofer committed 1 year ago

Verified db216443

Remove old build scripts (#532)

charleshofer committed 1 year ago

Verified d9b4b407

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

gulsumgudukbay committed 1 year ago

Verified 4ee8db98

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

gulsumgudukbay committed 1 year ago

Verified c07c3a15

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

gulsumgudukbay committed 1 year ago

Verified 68d7228c

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

gulsumgudukbay committed 1 year ago

Verified f697dde0

unskip pallas tests that were skipped due to LLVm error

gulsumgudukbay committed 1 year ago

41a32e35

addressing Charlie's comments

gulsumgudukbay committed 1 year ago

248cf432

removed a print stmt

gulsumgudukbay committed 1 year ago

e1236c74

Adding abort test info in run_single_gpu.py logs

gulsumgudukbay committed 1 year ago

1ff07701

fix is_device_rocm function call

root committed 1 year ago

bcc84983

Fix GPU handle pool singleton aliasing (#525)

phambinhfin committed 1 year ago

Verified 9890a51d

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

Arech8 committed 1 year ago

Verified 2c4c7064

adding additional skips only for MI350 for LLVM cannot scalarize ERROR

root committed 1 year ago

e5fc666b

Bugfix bazel debug rule (#511)

Arech8 committed 1 year 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 1 year ago

3d6b521d

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

Arech8 committed 1 year ago

Verified 03684216

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

Arech8 committed 1 year ago

58a5c560

Revert "unskip test_select_n"

gulsumgudukbay committed 1 year ago

81ba3269

skip test_fused_attention Interpret mode

gulsumgudukbay committed 1 year ago

5eed81de

unskip test_select_n

gulsumgudukbay committed 1 year ago

8bf16ba1

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

zahiqbal committed 1 year ago

7acefb20

skip more tests

gulsumgudukbay committed 1 year ago

5271d8f5

gulsumgudukbay committed 1 year ago

6d1f1ac1

Update scipy_signal_test.py

gulsumgudukbay committed 1 year ago

e1e0e2ed

skip additional tests

gulsumgudukbay committed 1 year ago

2ce571c9

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

gulsumgudukbay committed 1 year ago

f12689aa

[ROCM] Add missing dependency on amdhip64 runtime

mrodden committed 1 year ago

d9047090

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

mrodden committed 1 year ago

3a6435f6

Older