ROCm/jax

Pull Requests Commits

update to fix lint for pre-commit run

cj401-amd committed 2 days ago

4e144179

update with a simplified tests for rocm profiling

cj401-amd committed 2 days ago

af2fe9d1

re-format with 2-space indent

cj401-amd committed 2 days ago

5df0c0a3

update test for rocm profiling

cj401-amd committed 2 days ago

a70086c7

update unit test with kernel_details

cj401-amd committed 2 days ago

799aea57

Address Gemini review

JehandadKhan committed 2 days ago

50a524f9

[ROCm] update to test if there are GPU events when doing profiling on ROCm via matmul

cj401-amd committed 2 days ago

54ccbc15

Merge pull request #35442 from mattjj:fix-ugly-code

Google-ML-Automation committed 2 days ago

e3b1f068

Merge pull request #35450 from jakevdp:pyrefly-core

Google-ML-Automation committed 2 days ago

a39b86f2

Merge pull request #35532 from jakevdp:argsort-type

Google-ML-Automation committed 2 days ago

9cff4d4d

[typing] fix pyi type signature for argsort

jakevdp committed 2 days ago

926bf973

[pyrefly] fix pyrefly errors in jax._src.core

jakevdp committed 2 days ago

b80f8e82

Merge pull request #35522 from jax-ml:dependabot/github_actions/actions/checkout-6.0.2

Google-ML-Automation committed 2 days ago

23d35a9b

Merge pull request #35518 from jax-ml:dependabot/github_actions/actions/upload-artifact-7.0.0

Google-ML-Automation committed 2 days ago

1cc17554

[Mosaic:GPU] Fix the barrier value creation.

PatriosTheGreat committed 2 days ago

13792508

Skip installing collecting profile requirements under 3.13-nogil.

danielsuo committed 2 days ago

2260303e

Skip lax_numpy_reducers_test reducer tests with where.

danielsuo committed 2 days ago

6bbd83d9

Merge pull request #35497 from superbobry:pyrefly

Google-ML-Automation committed 2 days ago

81bb26ae

[NFC] Remove and modify some out-of-date test skips.

yueshengys committed 2 days ago

c0c98433

Bump actions/checkout from 6.0.0 to 6.0.2

dependabot[bot] committed 2 days ago

Verified a4dd1ddc

Bump actions/upload-artifact from 6.0.0 to 7.0.0

dependabot[bot] committed 2 days ago

Verified 1bd107b6

Add TPU v7 to `test_vmem_oom_error_message_basics`.

yueshengys committed 2 days ago

c097ebed

[Mosaic GPU] Add support for s8 and u8 warp-level MMAs

apaszke committed 2 days ago

3db21b72

Merge pull request #35453 from jakevdp:pyrefly-ad

Google-ML-Automation committed 2 days ago

4b423a65

[Mosaic GPU] Add support for untiled CP_ASYNC

apaszke committed 2 days ago

a1b958ba

Merge pull request #35417 from joshuapjacob:patch-1

Google-ML-Automation committed 2 days ago

a192b434

Merge pull request #34702 from ROCm:add-run-bazel-test-rocm-rbe

Google-ML-Automation committed 2 days ago

ba54b343

Reverts 41ae1eb6ea50739feb91589705aa078b1495d40f

hawkinsp committed 2 days ago

68b2565e

Merge pull request #35501 from jax-ml:postrelease/0.9.1

Google-ML-Automation committed 2 days ago

edf82d10

[Mosaic GPU][NFC] Remove dead `constraints.BroadcastInDim`.

allanrenucci committed 2 days ago

ed6d58f2

Older