Sparse Tensor Dialect は MLIR リポジトリにあります。
ヘッダー ファイル
ライブラリと C++ ファイル
- mlir/lib/Dialect/SparseTensor
- 実行エンジンのスパース テンソル ランタイム サポート ライブラリ
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/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 バインディング テスト)