稀疏張量方言存放區結構

Sparse Tensor Dialect 位於 MLIR 存放區。

標頭檔案

• mlir/include/mlir/Dialect/SparseTensor
• mlir/include/mlir/ExecutionEngine/SparseTensor

程式庫和 C++ 檔案

• mlir/lib/Dialect/SparseTensor
• 執行引擎適用的稀疏張量執行階段支援程式庫
  • mlir/lib/ExecutionEngine/SparseTensor (Runtime)

Python 繫結和 CAPI

• mlir/include/mlir-c/Dialect/SparseTensor.h
• mlir/lib/Bindings/Python/DialectSparseTensor.cpp (Passes)
• mlir/lib/CAPI/Dialect/SparseTensor.cpp (Passes)
• mlir/python/mlir/dialects/sparse_tensor.py

MLIR 基準

• mlir/benchmark/python/benchmark_sparse.py

測試命名空間

• mlir/test/Dialect/SparseTensor (FileCheck 測試)
• mlir/test/Integration/Dialect/SparseTensor (端對端測試)
• mlir/unittests/Dialect/SparseTensor (單元測試)
• mlir/test/CAPI/sparse_tensor.c (CAPI 測試)
• mlir/test/python/dialects/sparse_tensor/dialect.py (Python 繫結測試)