Unsupported ROCm Tests `tests/pallas/gpu_pallas_distributed_test.py`

[JehandadKhan]

Description

The tests in tests/pallas/gpu_pallas_distributed_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/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather_auto_vec_size_int0
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather_auto_vec_size_int1
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather_auto_vec_size_int2
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather_auto_vec_size0
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather_auto_vec_size1
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather_auto_vec_size2
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather_different_axes0
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather_different_axes1
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather_large_minor_dims
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather0
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather1
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather2
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather3
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather4
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_gather5
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_reduce0
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_reduce1
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_all_reduce2
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce0
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce1
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce10
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce11
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce12
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce13
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce14
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce15
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce16
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce17
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce18
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce19
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce2
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce20
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce3
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce4
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce5
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce6
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce7
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce8
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_load_reduce9
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_store_regs
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_multimem_store_tma
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter_auto_vec_size_int0
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter_auto_vec_size_int1
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter_auto_vec_size_int2
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter_auto_vec_size0
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter_auto_vec_size1
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter_auto_vec_size2
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter_different_axes0
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter_different_axes1
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter_large_minor_dims
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter0
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter1
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter2
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter3
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter4
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallMultimemTest:::test_reduce_scatter5
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallRemoteDMATest:::test_permuted_mesh
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallRemoteDMATest:::test_remote_dma_2d_mesh0
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallRemoteDMATest:::test_remote_dma_2d_mesh1
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallRemoteDMATest:::test_remote_dma_basic
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallRemoteDMATest:::test_semaphore_signal_collective_axes
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallRemoteDMATest:::test_signal_parallel
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallRemoteDMATest:::test_wait_nodec
• tests/pallas/gpu_pallas_distributed_test.py:::PallasCallRemoteDMATest:::test_wait_twice

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

JAX version 0.8.0
MI300 Accelerator
Independent of OS and Python version