ROCm/jax

Pull Requests Commits

skip more tests

gulsumgudukbay committed 1 year ago

310c6b38

gulsumgudukbay committed 1 year ago

32c7332c

Update scipy_signal_test.py

gulsumgudukbay committed 1 year ago

Verified a1f69826

skip additional tests

gulsumgudukbay committed 1 year ago

764f644c

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

Update tests/lax_numpy_test.py

gulsumgudukbay committed 1 year ago

c8411cc7

Update tests/experimental_rnn_test.py

gulsumgudukbay committed 1 year ago

f005d96c

[0.6.0-UT] skip failing tests

gulsumgudukbay committed 1 year ago

5e534a5b

Remove warnings.warn from get_rocm_version

psanal35 committed 1 year ago

Verified f95ad1ab

Add missing header `warnings` for test_util.py (#458)

psanal35 committed 1 year ago

Verified 7be899a0

Fix get_rocm_version to skip tests if installation is missing

psanal35 committed 1 year ago

Verified 3820e7ee

Older