ROCm/jax

Pull Requests Commits

force prevent_cse true

Ruturaj4 committed 190 days ago

45ada603

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

gulsumgudukbay committed 194 days ago

Verified 4ee8db98

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

gulsumgudukbay committed 195 days ago

Verified c07c3a15

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

gulsumgudukbay committed 196 days ago

Verified 68d7228c

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

gulsumgudukbay committed 196 days ago

Verified f697dde0

unskip pallas tests that were skipped due to LLVm error

gulsumgudukbay committed 196 days ago

41a32e35

addressing Charlie's comments

gulsumgudukbay committed 202 days ago

248cf432

removed a print stmt

gulsumgudukbay committed 202 days ago

e1236c74

Adding abort test info in run_single_gpu.py logs

gulsumgudukbay committed 202 days ago

1ff07701

fix is_device_rocm function call

root committed 205 days ago

bcc84983

Fix GPU handle pool singleton aliasing (#525)

phambinhfin committed 209 days ago

Verified 9890a51d

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

Arech8 committed 210 days ago

Verified 2c4c7064

adding additional skips only for MI350 for LLVM cannot scalarize ERROR

root committed 211 days ago

e5fc666b

Bugfix bazel debug rule (#511)

Arech8 committed 212 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 225 days ago

3d6b521d

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

Arech8 committed 230 days ago

Verified 03684216

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

Arech8 committed 231 days ago

58a5c560

Revert "unskip test_select_n"

gulsumgudukbay committed 232 days ago

81ba3269

skip test_fused_attention Interpret mode

gulsumgudukbay committed 232 days ago

5eed81de

unskip test_select_n

gulsumgudukbay committed 234 days ago

8bf16ba1

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

zahiqbal committed 240 days ago

7acefb20

skip more tests

gulsumgudukbay committed 241 days ago

5271d8f5

gulsumgudukbay committed 241 days ago

6d1f1ac1

Update scipy_signal_test.py

gulsumgudukbay committed 241 days ago

e1e0e2ed

skip additional tests

gulsumgudukbay committed 241 days ago

2ce571c9

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

gulsumgudukbay committed 249 days ago

f12689aa

[ROCM] Add missing dependency on amdhip64 runtime

mrodden committed 249 days ago

d9047090

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

mrodden committed 249 days ago

3a6435f6

Use HIPBLAS_V2 (#222)

Ruturaj4 committed 249 days ago

08d6cca2

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError (#465)

gulsumgudukbay committed 261 days ago

Verified c66db162

Older