Add ROCm benchmark workflow for MaxText #755
psanal35
wants to merge
50 commits into
amd-main
from
add-rocm-model-benchmarks
Remove nvidia_wheel_versions
43c0570e
Make jaxlib targets visible
bcef89c3
hipblas typedef fix
733b7bf8
No GPU fail
793d3127
Wrap HIP inline functions in anonymous namespaces in vendor.h
e3ad0ecb
SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError
a831ef20
Add shared utility function get_rocm_version to test_util.py
58249a4a
Fix hipSparse CSR algorithm mappings for ROCm 7
e587f903
Fix v_pages quantization and adjust test params for ROCm compatibilit…
80899473
Address LLVM assertion failure due to a multithreaded use. Update .gi…
d9e7020e
Add skip of test_is_finite() on Cuda (#565)
42a3be64
Add rocm test requirements file (#570)
544c6d4a
Let the unit tests use build.py for setting up Bazel commands for uni…
4673584d
adding abort logic to rocm/jax (#590)
1c79814a
Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (…
9b5d7088
Fix shared memory limit check for ROCm in test_dot (#596)
82bf13e1
Fix Numpy signatures test (#598)
ad47e174
fix merge arts
3b3b31cf
Enable RngShardingTests (#644)
8a9adefd
Enable test_variadic_reduce_window on ROCm (#647)
4eb74735
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
f360e135
Update sparse test skip messages in v0.8.2 (#653)
81842f41
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
489fcf60
Update sparse test skip messages in v0.8.2 (#653)
c2ea7b45
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
82a1e816
Update sparse test skip messages in v0.8.2 (#653)
2c12a03c
Enable testMultivariateNormalSingularCovariance on ROCm (#666)
5c681b10
Update Skip Reason Outputs (#663)
3757b64d
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
7ec9fe0f
Update sparse test skip messages in v0.8.2 (#653)
411d4faa
Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)
130ca422
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
837654d2
Update sparse test skip messages in v0.8.2 (#653)
2b8c7fe3
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
d0a11b30
Update sparse test skip messages in v0.8.2 (#653)
3e79165b
Remove 'mean' from unsupported params for jnp.var (#689)
9d4fce1a
Skipping testEighTinyNorm due to hipSolver issues (#697)
7601b868
Abort detection CI workflow (#688)
c21c3b8d
Abort-Detection: Fix halt-for-connection input (#712)
3f5828e9
fix: add rocm_sysdeps/lib to wheel RUNPATH (#737)
6283bbcf
Temporarily disable the cron trigger for the cont. wheel tests workflow
2422c280
Add placeholder for nightly benchmark workflow (#768)
8d4fbef1
Rename the benchmarks workflow consistently (#770)
eb58ba29
Add ROCm benchmark workflow for MaxText
2483b469
Resolve latest MaxText Transformer Engine wheel from ROCm MaxText rel…
1d25b87d
Load MaxText ROCm benchmark configs and requirements from ROCm/maxtext
24fb105b
Revisit ROCm benchmark results and run-manifest collection
34b2d0f8
Revisit ROCm artifact upload to S3 for reusability
37fe3f7a
Remove TE installation to keep the model lightweight
9183f204
Update benchmark target scripts for more generic use cases
182dd2a8
Login to write a write a comment.
Login via GitHub