Add ROCm benchmark workflow for MaxText#755

Closed

psanal35 wants to merge 50 commits into

add-rocm-model-benchmarks

Commits on May 6, 2026

Remove nvidia_wheel_versions

charleshofer
authored and
mminutoli
committed
Make jaxlib targets visible

charleshofer
authored and
mminutoli
committed
hipblas typedef fix

charleshofer
authored and
mminutoli
committed
No GPU fail

charleshofer
authored and
mminutoli
committed
Wrap HIP inline functions in anonymous namespaces in vendor.h
mminutoli
committed
SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd
authored and
mminutoli
committed
Add shared utility function get_rocm_version to test_util.py

charleshofer
authored and
mminutoli
committed
Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin
authored and
mminutoli
committed
Fix v_pages quantization and adjust test params for ROCm compatibilit… (#560)

phambinhfin
authored and
mminutoli
committed
Address LLVM assertion failure due to a multithreaded use. Update .gitignore (#563)

Arech8
authored and
mminutoli
committed
Add skip of test_is_finite() on Cuda (#565)

Arech8
authored and
mminutoli
committed
Add rocm test requirements file (#570)

AratiGanesh
authored and
mminutoli
committed
Let the unit tests use build.py for setting up Bazel commands for unit tests (#582)

charleshofer
authored and
mminutoli
committed
adding abort logic to rocm/jax (#590)

gulsumgudukbay
authored and
mminutoli
committed
Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (#597)

phambinhfin
authored and
mminutoli
committed
Fix shared memory limit check for ROCm in test_dot (#596)

phambinhfin
authored and
mminutoli
committed
Fix Numpy signatures test (#598)

authored and
mminutoli
committed
fix merge arts

Ruturaj4
authored and
mminutoli
committed
Enable RngShardingTests (#644)

gulsumgudukbay
authored and
mminutoli
committed
Enable test_variadic_reduce_window on ROCm (#647)
mminutoli
committed
Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd
authored and
mminutoli
committed
Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd
authored and
mminutoli
committed
Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd
authored and
mminutoli
committed
Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd
authored and
mminutoli
committed
Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd
authored and
mminutoli
committed
Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd
authored and
mminutoli
committed
Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh
authored and
mminutoli
committed
Update Skip Reason Outputs (#663)

gulsumgudukbay
authored and
mminutoli
committed
Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd
authored and
mminutoli
committed
Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd
authored and
mminutoli
committed
Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd
authored and
mminutoli
committed
Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd
authored and
mminutoli
committed
Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd
authored and
mminutoli
committed
Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd
authored and
mminutoli
committed
Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd
authored and
mminutoli
committed
Remove 'mean' from unsupported params for jnp.var (#689)

magaonka-amd
authored and
mminutoli
committed
Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh
authored and
mminutoli
committed
Abort detection CI workflow (#688)

gulsumgudukbay
authored and
mminutoli
committed
Abort-Detection: Fix halt-for-connection input (#712)

gulsumgudukbay
authored and
mminutoli
committed
fix: add rocm_sysdeps/lib to wheel RUNPATH (#737)

WBobby
authored and
mminutoli
committed
Temporarily disable the cron trigger for the cont. wheel tests workflow

psanal35
authored and
mminutoli
committed

Commits on May 11, 2026

Commits on May 13, 2026

Remove TE installation to keep the model lightweight
psanal35
committed

Commits on May 15, 2026

Update benchmark target scripts for more generic use cases
psanal35
committed