ROCm/jax

Pull Requests Commits

gulsumgudukbay committed 212 days ago

6d1f1ac1

Update scipy_signal_test.py

gulsumgudukbay committed 212 days ago

e1e0e2ed

skip additional tests

gulsumgudukbay committed 212 days ago

2ce571c9

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

gulsumgudukbay committed 219 days ago

f12689aa

[ROCM] Add missing dependency on amdhip64 runtime

mrodden committed 219 days ago

d9047090

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

mrodden committed 219 days ago

3a6435f6

Use HIPBLAS_V2 (#222)

Ruturaj4 committed 219 days ago

08d6cca2

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

gulsumgudukbay committed 232 days ago

Verified c66db162

added bidirectional condition back as they are passing

gulsumgudukbay committed 236 days ago

4952ad41

the tests are passing, unskip them

gulsumgudukbay committed 236 days ago

ab5203fd

Update linalg_test.py

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

8bdaf329

[0.6.0-UT] skip failing tests

gulsumgudukbay committed 236 days ago

86c380c7

gulsumgudukbay committed 236 days ago

cbdfa577

additional final skips from run_single_gpu.py

gulsumgudukbay committed 236 days ago

109c0a06

Update nn_test fix linting issue

gulsumgudukbay committed 236 days ago

1bfc74f5

Update nn_test.py to add testDotProductAttention21 skip

gulsumgudukbay committed 236 days ago

e8a07b27

skipped additional tests and fixed testSVD typo

gulsumgudukbay committed 236 days ago

8e281478

Update linalg_test.py

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

aaadb0f7

Account for versioned clang binaries

charleshofer committed 236 days ago

34c77988

Update tests/lax_numpy_test.py

gulsumgudukbay committed 236 days ago

c8411cc7

Update tests/experimental_rnn_test.py

gulsumgudukbay committed 236 days ago

f005d96c

[0.6.0-UT] skip failing tests

gulsumgudukbay committed 236 days ago

5e534a5b

Remove warnings.warn from get_rocm_version

psanal35 committed 236 days ago

Verified f95ad1ab

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

psanal35 committed 236 days ago

Verified 7be899a0

Fix get_rocm_version to skip tests if installation is missing

psanal35 committed 237 days ago

Verified 3820e7ee

updated the XLA commit hash

gulsumgudukbay committed 237 days ago

4694374d

Newer Older