Sparse Tensor Dialect si trova nel repository MLIR.
File di intestazione
Libreria e file C++
- mlir/lib/Dialect/SparseTensor
- Libreria di supporto del runtime del tensore sparso per il motore di esecuzione
Binding Python e 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
Benchmark MLIR
Test
- mlir/test/Dialect/SparseTensor (test FileCheck)
- mlir/test/Integration/Dialect/SparseTensor (test end-to-end)
- mlir/unittests/Dialect/SparseTensor (test delle unità)
- mlir/test/CAPI/sparse_tensor.c (test CAPI)
- mlir/test/python/dialects/sparse_tensor/dialect.py (test dei binding Python)