1 #pragma once 2 3 #include <ATen/Tensor.h> 4 #include <ATen/core/Scalar.h> 5 6 /* 7 Functions here use deprecated cuSPARSE API that was removed in CUDA 11. 8 Here only 32-bit indices sparse indices are supported. 9 This file will be removed eventually. 10 */ 11 12 namespace at::native { 13 14 void s_addmm_out_csr_sparse_dense_cuda_worker(int64_t nnz, int64_t m, int64_t n, int64_t k, const Tensor& r_, const Scalar& beta, const Tensor& t, const Scalar& alpha, const Tensor& crow_indices, const Tensor& col_indices, const Tensor& values, const Tensor& dense); 15 16 } // namespace at::native 17