ROCm/jax

Pull Requests Commits

Enabled ToeplitzSymmetricConstruction unit tests for ROCm devices (#608)

tsrw2048 committed 21 days ago

455f6b10

Enabled testTridiagonal to run on ROCm devices (#607)

tsrw2048 committed 21 days ago

ee2c254e

Ruturaj4 committed 21 days ago

e8ac89ba

Fixed merge conflicts when moving GESVDJ commit from JAX 0.8.0 to 0.8.2 (#602)

tsrw2048 committed 21 days ago

aaac58cb

Fix GPU lowering rule for SVD on ROCm devices (#600)

tsrw2048 committed 21 days ago

fd0e42c9

Fix Numpy signatures test (#598)

magaonka-amd committed 21 days ago

1a823af2

Fix shared memory limit check for ROCm in test_dot (#596)

phambinhfin committed 21 days ago

e5b66138

Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (#597)

phambinhfin committed 21 days ago

17c7560c

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 21 days ago

a83cc9e9

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

charleshofer committed 21 days ago

6bd1a15e

Add rocm test requirements file (#570)

AratiGanesh committed 21 days ago

835856e0

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

Arech8 committed 21 days ago

6ab35026

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

Arech8 committed 21 days ago

8b0b3d10

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

phambinhfin committed 21 days ago

61d28322

Make nvidia version data optional for ROCm builds

phambinhfin committed 21 days ago

d88ca192

Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin committed 21 days ago

81cdeb49

Add shared utility function get_rocm_version to test_util.py

charleshofer committed 21 days ago

1bac5e70

SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd committed 21 days ago

224b5baf

Wrap HIP inline functions in anonymous namespaces in vendor.h

AratiGanesh committed 21 days ago

2b08a20d

charleshofer committed 21 days ago

4f3741b3

hipblas typedef fix

charleshofer committed 21 days ago

6e0eb3ec

Make jaxlib targets visible

charleshofer committed 21 days ago

de91c59a

Remove nvidia_wheel_versions

charleshofer committed 21 days ago

c706c2fa

Use maxsize=None with trace_to_jaxpr's weakref_lru_cache to get more cache hits. This also restores previous behavior of lu.cache.

yashk2810 committed 24 days ago

2de5b8b0

Remove failing test_itof_dot_canonicalization_fails_without_compat_mode test.

18praveenb committed 28 days ago

98d6b4ba

Disable `tpu_splash_attention_kernel_test` on TPU v7x.

mwhittaker committed 28 days ago

28799d51

Add libtpu guard to failing tpu_trace_value_test.

mwhittaker committed 28 days ago

80b1ef6a

Skip tpu_pallas_distributed_test on 7x.

mwhittaker committed 28 days ago

77d9ffb0

Add libtpu date guard for failing test.

mwhittaker committed 28 days ago

9e7b005e

Re-enable `socket_transfer_test` internally.

emilyfertig committed 28 days ago

18b34a12

Newer Older