ROCm/jax

Pull Requests Commits

Let the unit tests use build.py for setting up Bazel commands for unit tests (#582)

charleshofer committed 90 days ago

b64a5e40

Add rocm test requirements file (#570)

AratiGanesh committed 90 days ago

514fec32

Add skip of test_is_finite() on Cuda (#565)

Arech8 committed 90 days ago

b78fe9a6

Address LLVM assertion failure due to a multithreaded use. Update .gitignore (#563)

Arech8 committed 90 days ago

cb904b38

Fix v_pages quantization and adjust test params for ROCm compatibilit… (#560)

phambinhfin committed 90 days ago

cfddfa60

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 90 days ago

3128e0e7

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 90 days ago

6eee40dd

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 90 days ago

d8deb09f

Wrap HIP inline functions in anonymous namespaces in vendor.h

mminutoli committed 90 days ago

caf56b60

charleshofer committed 90 days ago

d0e32075

hipblas typedef fix

charleshofer committed 90 days ago

789df45d

Make jaxlib targets visible

charleshofer committed 90 days ago

c7ec794c

Remove nvidia_wheel_versions

charleshofer committed 90 days ago

2fe53184

Merge pull request #35453 from jakevdp:pyrefly-ad

Google-ML-Automation committed 90 days ago

4b423a65

[Mosaic GPU] Add support for untiled CP_ASYNC

apaszke committed 90 days ago

a1b958ba

Merge pull request #35417 from joshuapjacob:patch-1

Google-ML-Automation committed 90 days ago

a192b434

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

Google-ML-Automation committed 90 days ago

ba54b343

Reverts 41ae1eb6ea50739feb91589705aa078b1495d40f

hawkinsp committed 90 days ago

68b2565e

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

Google-ML-Automation committed 90 days ago

edf82d10

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

allanrenucci committed 90 days ago

ed6d58f2

Regenerate type stubs for nanobind v2.12.0.

hawkinsp committed 90 days ago

0ded2fed

Add serialization support for Mosaic GPU kernels.

khasanovaa committed 90 days ago

a3ffef5e

[Mosaic GPU] Fix for the latest LLVM version

apaszke committed 91 days ago

44736b92

[Mosaic] Add assembly format to tpu.delay_op

vsytch committed 91 days ago

31fc0bd5

[pallas:sc] Switched to using `pltpu.emit_pipeline` in the lowering

superbobry committed 91 days ago

41430d32

[Mosaic GPU][NFC] Use more precise type for `op` in `_optimization_barrier_constraint_system`.

allanrenucci committed 91 days ago

e20119d1

Merge pull request #35502 from superbobry:stubgen

Google-ML-Automation committed 91 days ago

b4e692ea

[Mosaic GPU][NFC] Remove duplicate layout inference rule for `vector.BroadcastOp`.

allanrenucci committed 91 days ago

0cd1592e

[mgpu] Minor cleanup in `try_cluster_cancel`.

chr1sj0nes committed 91 days ago

95d95e3d

[jaxlib] :normalize_stubs now also fixes improper ``get`` classmethods

superbobry committed 91 days ago

c5d55755

Newer Older