[mlir][sparse] remove sparse2sparse path in library (#69247)

This cleans up all external entry points that will have to deal with
non-permutations, making any subsequent refactoring much more local to
the lib files.

Author: aartbik

mlir/include/mlir/Dialect/SparseTensor/IR/Enums.h

@@ -145,7 +145,6 @@ enum class Action : uint32_t {
   kEmpty = 0,
   kEmptyForward = 1,
   kFromCOO = 2,
-  kSparseToSparse = 3,
   kFromReader = 4,
   kToCOO = 5,
   kPack = 7,

mlir/include/mlir/ExecutionEngine/SparseTensor/Storage.h

@@ -47,7 +47,6 @@ extern "C" {
 /// kEmpty          -               STS, empty
 /// kEmptyForward   -               STS, empty, with forwarding COO
 /// kFromCOO        COO             STS, copied from the COO source
-/// kSparseToSparse STS             STS, copied from the STS source
 /// kToCOO          STS             COO, copied from the STS source
 /// kPack           buffers         STS, from level buffers
 /// kSortCOOInPlace STS             STS, sorted in place

mlir/include/mlir/ExecutionEngine/SparseTensorRuntime.h

@@ -8,7 +8,6 @@
 add_mlir_library(MLIRSparseTensorRuntime
   File.cpp
   MapRef.cpp
-  NNZ.cpp
   Storage.cpp
 
   EXCLUDE_FROM_LIBMLIR

mlir/lib/ExecutionEngine/SparseTensor/CMakeLists.txt

@@ -44,21 +44,10 @@ SparseTensorStorageBase::SparseTensorStorageBase( // NOLINT
   }
 }
 
-// Helper macro for generating error messages when some
-// `SparseTensorStorage<P,I,V>` is cast to `SparseTensorStorageBase`
-// and then the wrong "partial method specialization" is called.
+// Helper macro for wrong "partial method specialization" errors.
 #define FATAL_PIV(NAME)                                                        \
   MLIR_SPARSETENSOR_FATAL("<P,I,V> type mismatch for: " #NAME);
 
-#define IMPL_NEWENUMERATOR(VNAME, V)                                           \
-  void SparseTensorStorageBase::newEnumerator(                                 \
-      SparseTensorEnumeratorBase<V> **, uint64_t, const uint64_t *, uint64_t,  \
-      const uint64_t *) const {                                                \
-    FATAL_PIV("newEnumerator" #VNAME);                                         \
-  }
-MLIR_SPARSETENSOR_FOREVERY_V(IMPL_NEWENUMERATOR)
-#undef IMPL_NEWENUMERATOR
-
 #define IMPL_GETPOSITIONS(PNAME, P)                                            \
   void SparseTensorStorageBase::getPositions(std::vector<P> **, uint64_t) {    \
     FATAL_PIV("getPositions" #PNAME);                                          \

mlir/lib/ExecutionEngine/SparseTensor/NNZ.cpp

@@ -131,13 +131,6 @@ extern "C" {
       return SparseTensorStorage<P, C, V>::newFromCOO(                         \
           dimRank, dimSizes, lvlRank, lvlTypes, dim2lvl, lvl2dim, coo);        \
     }                                                                          \
-    case Action::kSparseToSparse: {                                            \
-      assert(ptr && "Received nullptr for SparseTensorStorage object");        \
-      auto &tensor = *static_cast<SparseTensorStorageBase *>(ptr);             \
-      return SparseTensorStorage<P, C, V>::newFromSparseTensor(                \
-          dimRank, dimSizes, lvlRank, lvlSizes, lvlTypes, dim2lvl, lvl2dim,    \
-          dimRank, tensor);                                                    \
-    }                                                                          \
     case Action::kFromReader: {                                                \
       assert(ptr && "Received nullptr for SparseTensorReader object");         \
       SparseTensorReader &reader = *static_cast<SparseTensorReader *>(ptr);    \

mlir/lib/ExecutionEngine/SparseTensor/Storage.cpp

@@ -8795,7 +8795,6 @@ cc_library(
     srcs = [
         "lib/ExecutionEngine/SparseTensor/File.cpp",
         "lib/ExecutionEngine/SparseTensor/MapRef.cpp",
-        "lib/ExecutionEngine/SparseTensor/NNZ.cpp",
         "lib/ExecutionEngine/SparseTensor/Storage.cpp",
     ],
     hdrs = [