@@ -211,8 +211,8 @@ struct CusparseMatrixMultiplyOp {
211211 std::is_same_v<c10::BFloat16, scalar_t > ||
212212 std::is_same_v<float , scalar_t > ||
213213 std::is_same_v<double , scalar_t > ||
214- std::is_same_v <c10::complex <float >, scalar_t > ||
215- std::is_same_v <c10::complex <double >, scalar_t >,
214+ std::is_same <c10::complex <float >, scalar_t >::value ||
215+ std::is_same <c10::complex <double >, scalar_t >::value ,
216216 " cusparseSpGEMM only supports data type of half, bfloat16, float, double and complex float, double." );
217217 // SpGEMM Computation
218218 TORCH_CUDASPARSE_CHECK (cusparseSpGEMM_createDescr (&spgemmDesc));
@@ -673,8 +673,8 @@ void sparse_sparse_matmul_cuda_kernel(
673673 std::is_same_v<c10::BFloat16, scalar_t > ||
674674 std::is_same_v<float , scalar_t > ||
675675 std::is_same_v<double , scalar_t > ||
676- std::is_same_v <c10::complex <float >, scalar_t > ||
677- std::is_same_v <c10::complex <double >, scalar_t >,
676+ std::is_same <c10::complex <float >, scalar_t >::value ||
677+ std::is_same <c10::complex <double >, scalar_t >::value ,
678678 " sparse_sparse_matmul_cuda_kernel only supports data type of half, bfloat16, float, double and complex float, double." );
679679
680680 // older versions of cusparse on Windows segfault for complex128 dtype
0 commit comments