remove skipTest associated to hipsparse transpose

[kliegeois]

Once ROCm/rocm-libraries#4111 is merged and used by ROCm/jax CI, these skipTest can be removed.

[AratiGanesh]

@kliegeois

I tested the changes from PR ROCm/rocm-libraries#4111 by building hipsparse/rocsparse from the develop branch inside the container, but encountered a segmentation fault.

Environment - jaxlib 0.8.0+rocm7.2.0, Python 3.11.13, GPU gfx1201

Build steps
git clone https://github.com/ROCm/rocm-libraries
cd rocm-libraries
git checkout develop

cd /rocm-libraries/projects/rocsparse
sudo ./install.sh -dci --no-rocblas -a gfx1201

cd /rocm-libraries/projects/hipsparse
sudo ./install.sh -dci

Test with hipsparse 4.3.0
python -m pytest jax/tests/sparse_test.py::SparseObjectTest::test_matmul12 -v --tb=short

Result - Segmentation fault in csr_fromdense_gpu_lowering

Could you please share the environment you used for testing? It appears to be an ABI compatibility issue between the 4.3.0 build and tjaxlib version.

[AratiGanesh]

The tests pass as expected

root@8546edac53b3:~# /root/kim/run_tests.sh > /root/test_results.log 2>&1
cat /root/test_results.log
========== Running: test_csr_matmul_ad2 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 10.49s ==============================
========== Running: test_csr_matmul_ad3 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 8.23s ===============================
========== Running: test_csr_matmul_ad5 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 8.29s ===============================
========== Running: test_csr_matmul_ad8 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 8.23s ===============================
========== Running: test_csr_matvec0 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 6.28s ===============================
========== Running: test_csr_matvec4 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 6.23s ===============================
========== Running: test_csr_matvec5 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 6.30s ===============================
========== Running: test_csr_matvec9 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 6.23s ===============================
========== Running: test_csr_spmv0 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 6.51s ===============================
========== Running: test_csr_spmv2 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 6.75s ===============================
========== Running: test_csr_spmv4 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 6.76s ===============================
========== Running: test_csr_spmv5 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 6.73s ===============================
========== Running: test_csr_spmv7 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 6.71s ===============================
========== Running: test_csr_spmv9 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 6.72s ===============================
========== Running: test_matmul12 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 7.06s ===============================
========== Running: test_matmul13 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 7.14s ===============================
========== Running: test_matmul15 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 7.28s ===============================
========== Running: test_matmul18 ==========
Test session starting on GPU ?
============================= test session starts ==============================
platform linux -- Python 3.11.13, pytest-9.0.2, pluggy-1.6.0
rootdir: /root/rocm-jax/jax
configfile: pyproject.toml
plugins: json-report-1.5.0, html-4.1.1, hypothesis-6.150.0, metadata-3.1.1, rerunfailures-16.1, reportlog-1.0.0
collected 1 item

tests/sparse_test.py . [100%]

============================== 1 passed in 7.12s ===============================