スパース テンソル言語リポジトリの構造

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（パス）
• mlir/lib/CAPI/Dialect/SparseTensor.cpp（パス）
• 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 バインディング テスト）