Der Sparse Tensor Dialect befindet sich im MLIR-Repository.
Header-Dateien
Bibliothek und C++-Dateien
- mlir/lib/Dialect/SparseTensor
- Unterstützungsbibliothek für die Laufzeit von dünnbesetzten Tensoren für die Ausführungs-Engine
Python-Bindungen und 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-Benchmark
Tests
- mlir/test/Dialect/SparseTensor (FileCheck-Tests)
- mlir/test/Integration/Dialect/SparseTensor (End-to-End-Tests)
- mlir/unittests/Dialect/SparseTensor (Unit tests)
- mlir/test/CAPI/sparse_tensor.c (CAPI-Tests)
- mlir/test/python/dialects/sparse_tensor/dialect.py (Python-Bindungstests)