Unsupported ROCm Tests `tests/nn_test.py`

[JehandadKhan]

Description

The tests in tests/nn_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/nn_test.py:::NNFunctionsTest:::testDotProductAttention0
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttention10
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttention12
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttention14
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttention16
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttention18
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttention2
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttention20
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttention22
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttention4
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttention6
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttention8
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttentionBiasGradient0
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttentionBiasGradient1
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttentionBiasGradient2
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttentionBiasGradient3
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttentionMask0
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttentionMask1
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttentionMask2
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttentionMask3
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttentionMask4
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttentionMask5
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttentionMask6
• tests/nn_test.py:::NNFunctionsTest:::testDotProductAttentionMask7
• tests/nn_test.py:::NNFunctionsTest:::testScaledDotGeneral0
• tests/nn_test.py:::NNFunctionsTest:::testScaledDotGeneral1
• tests/nn_test.py:::NNFunctionsTest:::testScaledDotGeneral2
• tests/nn_test.py:::NNFunctionsTest:::testScaledDotGeneral3
• tests/nn_test.py:::NNFunctionsTest:::testScaledDotGeneral4
• tests/nn_test.py:::NNFunctionsTest:::testScaledDotGeneral5
• tests/nn_test.py:::NNFunctionsTest:::testScaledMatmul0
• tests/nn_test.py:::NNFunctionsTest:::testScaledMatmul1
• tests/nn_test.py:::NNFunctionsTest:::testScaledMatmul10
• tests/nn_test.py:::NNFunctionsTest:::testScaledMatmul11
• tests/nn_test.py:::NNFunctionsTest:::testScaledMatmul2
• tests/nn_test.py:::NNFunctionsTest:::testScaledMatmul3
• tests/nn_test.py:::NNFunctionsTest:::testScaledMatmul4
• tests/nn_test.py:::NNFunctionsTest:::testScaledMatmul5
• tests/nn_test.py:::NNFunctionsTest:::testScaledMatmul6
• tests/nn_test.py:::NNFunctionsTest:::testScaledMatmul7
• tests/nn_test.py:::NNFunctionsTest:::testScaledMatmul8
• tests/nn_test.py:::NNFunctionsTest:::testScaledMatmul9

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

JAX version 0.8.0
MI300 Accelerator
Independent of OS and Python version