ROCm/jax

Pull Requests Commits

Fix to trim the input to be divisible by num gpus in test shard UT

AratiGanesh committed 337 days ago

e7e847ad

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

Arech8 committed 356 days ago

58a5c560

Revert "unskip test_select_n"

gulsumgudukbay committed 356 days ago

81ba3269

skip test_fused_attention Interpret mode

gulsumgudukbay committed 357 days ago

5eed81de

unskip test_select_n

gulsumgudukbay committed 359 days 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

Use HIPBLAS_V2 (#222)

Ruturaj4 committed 1 year ago

08d6cca2

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

gulsumgudukbay committed 1 year ago

Verified c66db162

added bidirectional condition back as they are passing

gulsumgudukbay committed 1 year ago

4952ad41

the tests are passing, unskip them

gulsumgudukbay committed 1 year ago

ab5203fd

Update linalg_test.py

gulsumgudukbay committed 1 year ago

2e58cec0

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

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

8bdaf329

[0.6.0-UT] skip failing tests

gulsumgudukbay committed 1 year ago

86c380c7

gulsumgudukbay committed 1 year ago

cbdfa577

additional final skips from run_single_gpu.py

gulsumgudukbay committed 1 year ago

109c0a06

Update nn_test fix linting issue

gulsumgudukbay committed 1 year ago

1bfc74f5

Update nn_test.py to add testDotProductAttention21 skip

gulsumgudukbay committed 1 year ago

e8a07b27

skipped additional tests and fixed testSVD typo

gulsumgudukbay committed 1 year ago

8e281478

Update linalg_test.py

gulsumgudukbay committed 1 year ago

6fb91f56

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

gulsumgudukbay committed 1 year ago

011b5b41

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

aaadb0f7

Account for versioned clang binaries

charleshofer committed 1 year ago

34c77988

Older