ROCm/jax

Pull Requests Commits

give more appropriate warning message when using PGLE with rocprofiler-sdk

cj401-amd committed 154 days ago

0c17ba1d

remove @unittest.skip('Test causes OOMs') as rocprofiler-sdk has no such issues anymore

cj401-amd committed 154 days ago

ec929e01

uncomment test causes OOMs

cj401-amd committed 154 days ago

4a90d342

add rocprofiler_configure as a global

cj401-amd committed 202 days ago

7fb62272

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

gulsumgudukbay committed 212 days ago

Verified c66db162

added bidirectional condition back as they are passing

gulsumgudukbay committed 217 days ago

4952ad41

the tests are passing, unskip them

gulsumgudukbay committed 217 days ago

ab5203fd

Update linalg_test.py

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

8bdaf329

[0.6.0-UT] skip failing tests

gulsumgudukbay committed 217 days ago

86c380c7

gulsumgudukbay committed 217 days ago

cbdfa577

additional final skips from run_single_gpu.py

gulsumgudukbay committed 217 days ago

109c0a06

Update nn_test fix linting issue

gulsumgudukbay committed 217 days ago

1bfc74f5

Update nn_test.py to add testDotProductAttention21 skip

gulsumgudukbay committed 217 days ago

e8a07b27

skipped additional tests and fixed testSVD typo

gulsumgudukbay committed 217 days ago

8e281478

Update linalg_test.py

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

aaadb0f7

Account for versioned clang binaries

charleshofer committed 217 days ago

34c77988

Update tests/lax_numpy_test.py

gulsumgudukbay committed 217 days ago

c8411cc7

Update tests/experimental_rnn_test.py

gulsumgudukbay committed 217 days ago

f005d96c

[0.6.0-UT] skip failing tests

gulsumgudukbay committed 217 days ago

5e534a5b

Remove warnings.warn from get_rocm_version

psanal35 committed 217 days ago

Verified f95ad1ab

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

psanal35 committed 217 days ago

Verified 7be899a0

Fix get_rocm_version to skip tests if installation is missing

psanal35 committed 217 days ago

Verified 3820e7ee

updated the XLA commit hash

gulsumgudukbay committed 217 days ago

4694374d

Build with ROCm 7.0 on new runner (#449)

charleshofer committed 217 days ago

Verified fbfeba5e

[0.6.0-UT] Fix NumpyLinalgTest test cases (#453)

phambinhfin committed 218 days ago

Verified 29c392d0

Select the platform GPUs for a few MultiProcessGpuTest test cases.

Pham Binh committed 218 days ago

f840dc0a

Older