Revert "[sparse] add extra options to _cslt_spare_mm (pytorch#137427)"

This reverts commit f145116. Reverted pytorch#137427 on behalf of https://github.com/huydhn due to This looks like the test is still failing, plz do a rebase ([comment](pytorch#137427 (comment)))
skreepshit · Nov 26, 2024 · 5318bf8 · 5318bf8
1 parent 6d4cd3e
commit 5318bf8
Show file tree

Hide file tree

Showing 8 changed files with 523 additions and 445 deletions.
diff --git a/aten/src/ATen/native/native_functions.yaml b/aten/src/ATen/native/native_functions.yaml
@@ -3371,7 +3371,7 @@
   dispatch:
     CUDA: _cslt_compress
 
-- func: _cslt_sparse_mm(Tensor compressed_A, Tensor dense_B, Tensor? bias=None, Tensor? alpha=None, ScalarType? out_dtype=None, bool transpose_result=False, int alg_id=0, int split_k=1, bool split_k_one_kernel=True) -> Tensor
+- func: _cslt_sparse_mm(Tensor compressed_A, Tensor dense_B, Tensor? bias=None, Tensor? alpha=None, ScalarType? out_dtype=None, bool transpose_result=False, int alg_id=0) -> Tensor
   dispatch:
     CUDA: _cslt_sparse_mm