El dialecto de tensores dispersos se encuentra en el repositorio de MLIR.
Archivos de encabezado
Archivos de biblioteca y C++
- mlir/lib/Dialect/SparseTensor
- Biblioteca de compatibilidad con el entorno de ejecución de tensores dispersos para el motor de ejecución
Vinculaciones de Python y CAPI
- mlir/include/mlir-c/Dialect/SparseTensor.h
- mlir/lib/Bindings/Python/DialectSparseTensor.cpp (Pases)
- mlir/lib/CAPI/Dialect/SparseTensor.cpp (Pases)
- mlir/python/mlir/dialects/sparse_tensor.py
Comparativa de MLIR
Pruebas
- mlir/test/Dialect/SparseTensor (pruebas de FileCheck)
- mlir/test/Integration/Dialect/SparseTensor (pruebas de extremo a extremo)
- mlir/unittests/Dialect/SparseTensor (pruebas de unidades)
- mlir/test/CAPI/sparse_tensor.c (pruebas de CAPI)
- mlir/test/python/dialects/sparse_tensor/dialect.py (pruebas de vinculación de Python)