Unsupported ROCm Tests `tests/scaled_matmul_stablehlo_test.py`

[JehandadKhan]

Description

The tests in tests/scaled_matmul_stablehlo_test.py are being skipped and need to be triaged to assess if they can be enabled. If they cant due to missing ROCm features, then we need to implement such a feature.

• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_nvfp40
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_nvfp41
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_nvfp42
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_nvfp43
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_nvfp44
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_nvfp45
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_nvfp46
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_nvfp47
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_nvfp48
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_nvfp49
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_sharded0
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_sharded1
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_sharded2
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_sharded3
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_sharded4
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_sharded5
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_sharded6
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_sharded7
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_vmap0
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_vmap1
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general_vmap2
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general0
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general1
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general2
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general3
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general4
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general5
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general6
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general7
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general8
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_dot_general9
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_nvfp4_gradient_clip0
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_nvfp4_gradient_clip1
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_nvfp4_gradient_clip2
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_nvfp4_gradient_clip3
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_quantize_nvfp40
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_quantize_nvfp41
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_quantize_nvfp42
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_quantize_requires_global_scale0
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_quantize_requires_global_scale1
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_remat_checkpoint_dots
• tests/scaled_matmul_stablehlo_test.py:::ScaledDotGeneralTest:::test_remat_checkpoint_dots_with_no_batch_dims
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_collectives0
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_collectives1
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_collectives2
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_collectives3
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_collectives4
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_collectives5
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_collectives6
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_collectives7
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_nvfp40
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_nvfp41
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_nvfp42
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_nvfp43
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_nvfp44
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_nvfp45
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_nvfp46
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_nvfp47
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_nvfp48
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_nvfp49
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_sharded0
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_sharded1
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_sharded2
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_sharded3
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_sharded4
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_sharded5
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_sharded6
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul_sharded7
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul0
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul1
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul2
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul3
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul4
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul5
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul6
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul7
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul8
• tests/scaled_matmul_stablehlo_test.py:::ScaledMatmulTest:::test_scaled_matmul9

System info (python version, jaxlib version, accelerator, etc.)

JAX version 0.8.0
MI300 Accelerator
Independent of OS and Python version