ROCm/jax

Pull Requests Commits

phambinhfin committed 193 days ago

a7009f60

phambinhfin committed 193 days ago

0a2890b9

phambinhfin committed 193 days ago

4b82c983

phambinhfin committed 194 days ago

9fe456d9

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 209 days ago

3d6b521d

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

Arech8 committed 214 days ago

Verified 03684216

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

Arech8 committed 215 days ago

58a5c560

Revert "unskip test_select_n"

gulsumgudukbay committed 215 days ago

81ba3269

skip test_fused_attention Interpret mode

gulsumgudukbay committed 216 days ago

5eed81de

unskip test_select_n

gulsumgudukbay committed 217 days ago

8bf16ba1

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

zahiqbal committed 224 days ago

7acefb20

skip more tests

gulsumgudukbay committed 225 days ago

5271d8f5

gulsumgudukbay committed 225 days ago

6d1f1ac1

Update scipy_signal_test.py

gulsumgudukbay committed 225 days ago

e1e0e2ed

skip additional tests

gulsumgudukbay committed 225 days ago

2ce571c9

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

gulsumgudukbay committed 232 days ago

f12689aa

[ROCM] Add missing dependency on amdhip64 runtime

mrodden committed 232 days ago

d9047090

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

mrodden committed 232 days ago

3a6435f6

Use HIPBLAS_V2 (#222)

Ruturaj4 committed 232 days ago

08d6cca2

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

gulsumgudukbay committed 245 days ago

Verified c66db162

added bidirectional condition back as they are passing

gulsumgudukbay committed 249 days ago

4952ad41

the tests are passing, unskip them

gulsumgudukbay committed 249 days ago

ab5203fd

Update linalg_test.py

gulsumgudukbay committed 249 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 249 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 249 days ago

8bdaf329

[0.6.0-UT] skip failing tests

gulsumgudukbay committed 249 days ago

86c380c7

gulsumgudukbay committed 249 days ago

cbdfa577

additional final skips from run_single_gpu.py

gulsumgudukbay committed 249 days ago

109c0a06

Update nn_test fix linting issue

gulsumgudukbay committed 249 days ago

1bfc74f5

Update nn_test.py to add testDotProductAttention21 skip

gulsumgudukbay committed 249 days ago

e8a07b27

Older