ROCm/jax

Pull Requests Commits

Add HIP runtime dependency for hybird library

Pham Binh committed 260 days ago

b98df0a3

added bidirectional condition back as they are passing

gulsumgudukbay committed 260 days ago

4952ad41

the tests are passing, unskip them

gulsumgudukbay committed 260 days ago

ab5203fd

Update linalg_test.py

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

8bdaf329

[0.6.0-UT] skip failing tests

gulsumgudukbay committed 260 days ago

86c380c7

gulsumgudukbay committed 260 days ago

cbdfa577

additional final skips from run_single_gpu.py

gulsumgudukbay committed 260 days ago

109c0a06

Update nn_test fix linting issue

gulsumgudukbay committed 260 days ago

1bfc74f5

Update nn_test.py to add testDotProductAttention21 skip

gulsumgudukbay committed 260 days ago

e8a07b27

skipped additional tests and fixed testSVD typo

gulsumgudukbay committed 260 days ago

8e281478

Update linalg_test.py

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

aaadb0f7

Account for versioned clang binaries

charleshofer committed 260 days ago

34c77988

Update tests/lax_numpy_test.py

gulsumgudukbay committed 260 days ago

c8411cc7

Update tests/experimental_rnn_test.py

gulsumgudukbay committed 260 days ago

f005d96c

[0.6.0-UT] skip failing tests

gulsumgudukbay committed 260 days ago

5e534a5b

Remove warnings.warn from get_rocm_version

psanal35 committed 261 days ago

Verified f95ad1ab

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

psanal35 committed 261 days ago

Verified 7be899a0

Fix get_rocm_version to skip tests if installation is missing

psanal35 committed 261 days ago

Verified 3820e7ee

updated the XLA commit hash

gulsumgudukbay committed 261 days ago

4694374d

Build with ROCm 7.0 on new runner (#449)

charleshofer committed 261 days ago

Verified fbfeba5e

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

phambinhfin committed 262 days ago

Verified 29c392d0

Select the platform GPUs for a few MultiProcessGpuTest test cases.

Pham Binh committed 262 days ago

f840dc0a

Update rocm-test-requirements.txt

gulsumgudukbay committed 262 days ago

9e5c47cb

Revert "fixed typo"

gulsumgudukbay committed 262 days ago

69ec57e6

gulsumgudukbay committed 262 days ago

d2b14cb4

[0.6.0-UT] adding a QA test requirements file for the run_single_gpu test script… (#446)

gulsumgudukbay committed 263 days ago

Verified 1fc5d2a5

Older