pytorch
33da6c89 - [sparse] Add i8i8->i32 support for cuSPARSELt (#110499)

Commit
1 year ago
[sparse] Add i8i8->i32 support for cuSPARSELt (#110499) Summary: With the release of cuSPARSELt v0.5.0, we now have support for int8 int8 -> int32 matmul. This PR adds support for this via out_dtype. Test Plan: ``` python test/test_sparse_semi_structured.py -k int32 ``` Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/110499 Approved by: https://github.com/cpuhrsch
Author
Committer
Parents
Loading