O dialeto de tensor esparso está localizado no repositório MLIR (link em inglês).
Arquivos de cabeçalho
Arquivos de biblioteca e C++
- mlir/lib/Dialect/SparseTensor
- Biblioteca de suporte de tempo de execução de tensor esparso para mecanismo de execução
Vinculações do 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
Comparativo de mercado do MLIR
Testes
- mlir/test/Dialect/SparseTensor (testes do FileCheck)
- mlir/test/Integration/Dialect/SparseTensor (testes de ponta a ponta)
- mlir/unittests/Dialect/SparseTensor (testes de unidade)
- mlir/test/CAPI/sparse_tensor.c (testes de CAPI)
- mlir/test/python/dialects/sparse_tensor/dialect.py (testes de vinculação do Python)