[ROCm] enable complex128 in test_addmm_sizes_all_sparse_csr for rocm for trivial (k,n,m) cases

[dnikolaev-amd]

This PR enables test_addmm_sizes_all_sparse_csr_k_*_n_*_m_*_cuda_complex128 for ROCm for trivial cases (m or n or k = 0)

CUSPARSE_SPMM_COMPLEX128_SUPPORTED also used for test_addmm_all_sparse_csr and test_sparse_matmul and both of them are skipped for ROCm by @skipIfRocm or @skipCUDAIf(not _check_cusparse_spgemm_available())

cc: @jithunnair-amd

cc @jeffdaily @sunway513 @jithunnair-amd @pruthvistony @ROCmSupport @dllehr-amd @jataylo @hongxiayang

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/120504

• 📄 Preview Python docs built from this PR
• 📄 Preview C++ docs built from this PR
• ❓ Need help or want to give feedback on the CI? Visit the bot commands wiki or our office hours

Note: Links to docs will display an error until the docs builds have been completed.

✅ You can merge normally! (1 Unrelated Failure)

As of commit a89389d with merge base 605c0a2 ():

UNSTABLE - The following job failed but was likely due to flakiness present on trunk and has been marked as unstable:

• pull / linux-focal-py3_8-clang9-xla / test (xla, 1, 1, linux.12xlarge, unstable) (gh)
  test_xla_sharded_tensor_repr

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[pytorch-bot]

Please seek CI approval before scheduling CIFlow labels

[pytorch-bot]

Please seek CI approval before scheduling CIFlow labels

[pytorch-bot]

Please seek CI approval before scheduling CIFlow labels

[dnikolaev-amd]

@pytorchbot rebase

[pytorchmergebot]

@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here

[pytorchmergebot]

Successfully rebased dnikolaev/enable_test_addmm_sizes_all_sparse_csr_for_complex128_on_rocm_for_trivial_cases onto refs/remotes/origin/viable/strict, please pull locally before adding more changes (for example, via git checkout dnikolaev/enable_test_addmm_sizes_all_sparse_csr_for_complex128_on_rocm_for_trivial_cases && git pull --rebase)

[dnikolaev-amd]

@pytorchbot rebase

[pytorchmergebot]

@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here

[pytorchmergebot]

Successfully rebased dnikolaev/enable_test_addmm_sizes_all_sparse_csr_for_complex128_on_rocm_for_trivial_cases onto refs/remotes/origin/viable/strict, please pull locally before adding more changes (for example, via git checkout dnikolaev/enable_test_addmm_sizes_all_sparse_csr_for_complex128_on_rocm_for_trivial_cases && git pull --rebase)

[dnikolaev-amd]

@pytorchbot rebase

[pytorchmergebot]

@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here

[pytorchmergebot]

Successfully rebased dnikolaev/enable_test_addmm_sizes_all_sparse_csr_for_complex128_on_rocm_for_trivial_cases onto refs/remotes/origin/viable/strict, please pull locally before adding more changes (for example, via git checkout dnikolaev/enable_test_addmm_sizes_all_sparse_csr_for_complex128_on_rocm_for_trivial_cases && git pull --rebase)

[pytorchmergebot]

@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here

[pytorchmergebot]

Successfully rebased dnikolaev/enable_test_addmm_sizes_all_sparse_csr_for_complex128_on_rocm_for_trivial_cases onto refs/remotes/origin/viable/strict, please pull locally before adding more changes (for example, via git checkout dnikolaev/enable_test_addmm_sizes_all_sparse_csr_for_complex128_on_rocm_for_trivial_cases && git pull --rebase)

[dnikolaev-amd]

All tests passed. 19 tests enabled for ROCm #120504 (comment)

[jithunnair-amd]

@malfet Please review and merge this PR, it's enabling some trivial cases for sparse unit tests for ROCm6.0 and above. CI passed: #120504 (comment)

[dnikolaev-amd]

@pytorchbot rebase

[pytorchmergebot]

@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here

[pytorchmergebot]

Successfully rebased dnikolaev/enable_test_addmm_sizes_all_sparse_csr_for_complex128_on_rocm_for_trivial_cases onto refs/remotes/origin/viable/strict, please pull locally before adding more changes (for example, via git checkout dnikolaev/enable_test_addmm_sizes_all_sparse_csr_for_complex128_on_rocm_for_trivial_cases && git pull --rebase)

[dnikolaev-amd]

@pytorchbot rebase

[pytorchmergebot]

@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here

[pytorchmergebot]

Successfully rebased dnikolaev/enable_test_addmm_sizes_all_sparse_csr_for_complex128_on_rocm_for_trivial_cases onto refs/remotes/origin/viable/strict, please pull locally before adding more changes (for example, via git checkout dnikolaev/enable_test_addmm_sizes_all_sparse_csr_for_complex128_on_rocm_for_trivial_cases && git pull --rebase)

[ezyang]

@pytorchbot merge

[pytorchmergebot]

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging

Check the merge workflow status
here