CSR: Relaxing constraints to s_addmm_out_sparse_dense_cuda_worker

[aocsa]

🚀 Feature

Given that cusparse APIs support setting ldb (leading dimension of B).

pytorch/aten/src/ATen/native/sparse/cuda/SparseCUDATensorMath.cu

Line 83 in 1bd22e2

if(dense.stride(0) == 1 && dense.stride(1) == dense.size(0)) {

The previous condition is too strict condition as dense.stride(1) can be larger than dense.size(0). The first stride stride(1) can be size(0) * s where s is some positive integer.

Additional context

This function is used by addmm_out_sparse_csr_dense_cuda (CSR) and s_addmm_out_sparse_dense_cuda_worker (COO) tensor's operation.

Mentioned here:
#59012 (comment)

cc @aocsa @nikitaved @pearu @cpuhrsch @IvanYashchuk @mruberry