ROCm/jax

Pull Requests Commits

Update tests/lax_numpy_test.py

gulsumgudukbay committed 257 days ago

c8411cc7

Update tests/experimental_rnn_test.py

gulsumgudukbay committed 257 days ago

f005d96c

[0.6.0-UT] skip failing tests

gulsumgudukbay committed 257 days ago

5e534a5b

Remove warnings.warn from get_rocm_version

psanal35 committed 257 days ago

Verified f95ad1ab

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

psanal35 committed 257 days ago

Verified 7be899a0

Fix get_rocm_version to skip tests if installation is missing

psanal35 committed 257 days ago

Verified 3820e7ee

updated the XLA commit hash

gulsumgudukbay committed 257 days ago

4694374d

Build with ROCm 7.0 on new runner (#449)

charleshofer committed 257 days ago

Verified fbfeba5e

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

phambinhfin committed 258 days ago

Verified 29c392d0

Select the platform GPUs for a few MultiProcessGpuTest test cases.

Pham Binh committed 258 days ago

f840dc0a

Update rocm-test-requirements.txt

gulsumgudukbay committed 258 days ago

9e5c47cb

Revert "fixed typo"

gulsumgudukbay committed 258 days ago

69ec57e6

gulsumgudukbay committed 259 days ago

d2b14cb4

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

gulsumgudukbay committed 259 days ago

Verified 1fc5d2a5

[0.6.0-UT] Skip test_dot for ROCm <6.5, and shapes exceed shared memory limits (#448)

psanal35 committed 259 days ago

Verified d6456448

[0.6.0-UT] Add shared utility function get_rocm_version to test_util.py

psanal35 committed 260 days ago

Verified 893c907c

unskip test_select_n as it is passing on 0.6.0

gulsumgudukbay committed 264 days ago

5b38406a

added conditional seq_len assignment for lower dimension for tests/pallas/gpu_ops_test.py::FusedAttentionTest::test_fused_attention_[fwd,bwd] tests with the rationale. The test passes now because the input fits in shared memory. before, the tests were failing due to the seq exceeding the shared memory size and causing XLARuntimeError.

gulsumgudukbay committed 264 days ago

5659c086

[JAX] Fix numerical instability in testEighTinyNorm for ROCm GPU

phambinhfin committed 264 days ago

3c6ab9f3

JAX release v0.6.0

hawkinsp committed 292 days ago

9e9a3f03

[Pallas] Fix relative links in pipelining docs

justinjfu committed 292 days ago

28538e4c

Fix bug in argnums_partial_except when static_argnums is unsorted.

dfm committed 292 days ago

34234010

Return False in `is_env_present` if importing kubernetes leads to a ModuleNotFoundError

yashk2810 committed 292 days ago

8d3a92e6

Fix a bug in reduce_window sharding rule where padding is a tuple but we were checking for a scalar instead. Fixes https://github.com/jax-ml/jax/issues/28070

yashk2810 committed 292 days ago

33d43f7a

[Pallas] Move pipelining doc figures to static folder

justinjfu committed 292 days ago

a9465e70

Add explicit sharding to Introduction to parallel programming doc

yashk2810 committed 292 days ago

24fdc0ad

Update XLA dependency to use revision

Google-ML-Automation committed 292 days ago

6b94a5ce

[Pallas] Refactor pipelining docs.

justinjfu committed 292 days ago

17e2dcaf

Update job name to show "build only" on Linux ARM64.

nitins17 committed 292 days ago

bc4fcd87

disable packed layout test on old arch prior to Hopper

Cjkkkk committed 292 days ago

fb875101

Newer Older