ROCm/jax

Pull Requests Commits

Fix GPU handle pool singleton aliasing with tag-based template separation

phambinhfin committed 141 days ago

09f9ed01

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

gulsumgudukbay committed 173 days ago

6d1f1ac1

Update scipy_signal_test.py

gulsumgudukbay committed 173 days ago

e1e0e2ed

skip additional tests

gulsumgudukbay committed 173 days ago

2ce571c9

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

gulsumgudukbay committed 180 days ago

f12689aa

[ROCM] Add missing dependency on amdhip64 runtime

mrodden committed 180 days ago

d9047090

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

mrodden committed 180 days ago

3a6435f6

Use HIPBLAS_V2 (#222)

Ruturaj4 committed 180 days ago

08d6cca2

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

gulsumgudukbay committed 193 days ago

Verified c66db162

added bidirectional condition back as they are passing

gulsumgudukbay committed 197 days ago

4952ad41

the tests are passing, unskip them

gulsumgudukbay committed 197 days ago

ab5203fd

Update linalg_test.py

gulsumgudukbay committed 197 days ago

2e58cec0

remove ops_test skip function and skip call, Pakize will cherry-pick 0.5.0 PR as it is merged

gulsumgudukbay committed 197 days ago

5208fb5e

fixed linting stuff for jax/_src/lax/control_flow files, added more skips to the multiprocess_gpu_test.py, reduced the skipped tests for test_dot, used test_util get_rocm_version function instead of duplicating it

gulsumgudukbay committed 197 days ago

8bdaf329

[0.6.0-UT] skip failing tests

gulsumgudukbay committed 197 days ago

86c380c7

gulsumgudukbay committed 197 days ago

cbdfa577

additional final skips from run_single_gpu.py

gulsumgudukbay committed 197 days ago

109c0a06

Update nn_test fix linting issue

gulsumgudukbay committed 197 days ago

1bfc74f5

Update nn_test.py to add testDotProductAttention21 skip

gulsumgudukbay committed 197 days ago

e8a07b27

skipped additional tests and fixed testSVD typo

gulsumgudukbay committed 197 days ago

8e281478

Update linalg_test.py

gulsumgudukbay committed 197 days ago

6fb91f56

remove ops_test skip function and skip call, Pakize will cherry-pick 0.5.0 PR as it is merged

gulsumgudukbay committed 197 days ago

011b5b41

Older