pytorch
1d857586 - [ROCM] enable hipSOLVER backend for linalg.ldl_factor (#102665)

Commit
1 year ago
[ROCM] enable hipSOLVER backend for linalg.ldl_factor (#102665) * Add complex dtype support for linalg.ldl_factor * Fixes SWDEV-360139 * Enable the following 19 tests for ROCM + test_decomp.py TestDecompCUDA.test_comprehensive_linalg_ldl_factor_cuda_complex128 + test_decomp.py TestDecompCUDA.test_comprehensive_linalg_ldl_factor_cuda_complex64 + test_decomp.py TestDecompCUDA.test_comprehensive_linalg_ldl_factor_ex_cuda_complex128 + test_decomp.py TestDecompCUDA.test_comprehensive_linalg_ldl_factor_ex_cuda_complex64 + test_meta.py TestMetaCUDA.test_dispatch_meta_linalg_ldl_factor_cuda_complex128 + test_meta.py TestMetaCUDA.test_dispatch_meta_linalg_ldl_factor_cuda_complex64 + test_meta.py TestMetaCUDA.test_dispatch_meta_linalg_ldl_factor_ex_cuda_complex128 + test_meta.py TestMetaCUDA.test_dispatch_meta_linalg_ldl_factor_ex_cuda_complex64 + test_meta.py TestMetaCUDA.test_meta_linalg_ldl_factor_cuda_complex128 + test_ops.py TestCommonCUDA.test_noncontiguous_samples_linalg_ldl_factor_cuda_complex64 + test_ops.py TestCommonCUDA.test_noncontiguous_samples_linalg_ldl_factor_ex_cuda_complex64 + test_ops.py TestCommonCUDA.test_variant_consistency_eager_linalg_ldl_factor_cuda_complex64 + test_ops.py TestCommonCUDA.test_variant_consistency_eager_linalg_ldl_factor_ex_cuda_complex64 + test_ops.py TestMathBitsCUDA.test_conj_view_linalg_ldl_factor_cuda_complex64 + test_ops.py TestMathBitsCUDA.test_conj_view_linalg_ldl_factor_ex_cuda_complex64 + test_ops.py TestMathBitsCUDA.test_neg_conj_view_linalg_ldl_factor_cuda_complex128 + test_ops.py TestMathBitsCUDA.test_neg_conj_view_linalg_ldl_factor_ex_cuda_complex128 + test_ops_jit.py TestJitCUDA.test_variant_consistency_jit_linalg_ldl_factor_cuda_complex64 + test_ops_jit.py TestJitCUDA.test_variant_consistency_jit_linalg_ldl_factor_ex_cuda_complex64 Pull Request resolved: https://github.com/pytorch/pytorch/pull/102665 Approved by: https://github.com/lezcano
Author
Committer
Parents
Loading