'aten::mm' seems not be implemented for GPU?

RSY: <p>When I want to use the multiplication operator between dense matrix and sparse matrix (CSR format), I found that it does not implemented for GPU yet (it works for CPU). May I ask is there any technical problem that stops this implementation? and will it be implemented in the future? Thank you!</p>
<p>Here is the printout:</p>
<p>`NotImplementedError: Could not run ‘aten::mm’ with arguments from the ‘SparseCsrCUDA’ backend. This could be because the operator doesn’t exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit <a href="https://fburl.com/ptmfixes" class="inline-onebox" rel="noopener nofollow ugc">Internal Login</a> for possible resolutions. ‘aten::mm’ is only available for these backends: [CPU, CUDA, SparseCPU, SparseCUDA, SparseCsrCPU, BackendSelect, Named, ADInplaceOrView, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, UNKNOWN_TENSOR_TYPE_ID, AutogradMLC, AutogradHPU, AutogradNestedTensor, AutogradPrivateUse1, AutogradPrivateUse2, AutogradPrivateUse3, Tracer, Autocast, Batched, VmapMode].</p>
<p>CPU: registered at /pytorch/build/aten/src/ATen/RegisterCPU.cpp:16286 [kernel]<br>
CUDA: registered at /pytorch/build/aten/src/ATen/RegisterCUDA.cpp:20674 [kernel]<br>
SparseCPU: registered at /pytorch/build/aten/src/ATen/RegisterSparseCPU.cpp:959 [kernel]<br>
SparseCUDA: registered at /pytorch/build/aten/src/ATen/RegisterSparseCUDA.cpp:1019 [kernel]<br>
SparseCsrCPU: registered at /pytorch/build/aten/src/ATen/RegisterSparseCsrCPU.cpp:200 [kernel]<br>
BackendSelect: fallthrough registered at /pytorch/aten/src/ATen/core/BackendSelectFallbackKernel.cpp:3 [backend fallback]<br>
Named: fallthrough registered at /pytorch/aten/src/ATen/core/NamedRegistrations.cpp:11 [kernel]<br>
ADInplaceOrView: fallthrough registered at /pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:60 [backend fallback]<br>
AutogradOther: registered at /pytorch/torch/csrc/autograd/generated/VariableType_3.cpp:10820 [autograd kernel]<br>
AutogradCPU: registered at /pytorch/torch/csrc/autograd/generated/VariableType_3.cpp:10820 [autograd kernel]<br>
AutogradCUDA: registered at /pytorch/torch/csrc/autograd/generated/VariableType_3.cpp:10820 [autograd kernel]<br>
AutogradXLA: registered at /pytorch/torch/csrc/autograd/generated/VariableType_3.cpp:10820 [autograd kernel]<br>
UNKNOWN_TENSOR_TYPE_ID: registered at /pytorch/torch/csrc/autograd/generated/VariableType_3.cpp:10820 [autograd kernel]<br>
AutogradMLC: registered at /pytorch/torch/csrc/autograd/generated/VariableType_3.cpp:10820 [autograd kernel]<br>
AutogradHPU: registered at /pytorch/torch/csrc/autograd/generated/VariableType_3.cpp:10820 [autograd kernel]<br>
AutogradNestedTensor: registered at /pytorch/torch/csrc/autograd/generated/VariableType_3.cpp:10820 [autograd kernel]<br>
AutogradPrivateUse1: registered at /pytorch/torch/csrc/autograd/generated/VariableType_3.cpp:10820 [autograd kernel]<br>
AutogradPrivateUse2: registered at /pytorch/torch/csrc/autograd/generated/VariableType_3.cpp:10820 [autograd kernel]<br>
AutogradPrivateUse3: registered at /pytorch/torch/csrc/autograd/generated/VariableType_3.cpp:10820 [autograd kernel]<br>
Tracer: registered at /pytorch/torch/csrc/autograd/generated/TraceType_3.cpp:10078 [kernel]<br>
Autocast: registered at /pytorch/aten/src/ATen/autocast_mode.cpp:259 [kernel]<br>
Batched: registered at /pytorch/aten/src/ATen/BatchingRegistrations.cpp:1023 [kernel]<br>
VmapMode: fallthrough registered at /pytorch/aten/src/ATen/VmapModeRegistrations.cpp:33 [backend fallback]`</p>