Structure du dépôt du dialecte Tensor creux

Le dialecte Sparse Tensor se trouve dans le dépôt MLIR.

Fichiers d'en-tête

• mlir/include/mlir/Dialect/SparseTensor
• mlir/include/mlir/ExecutionEngine/SparseTensor

Bibliothèque et fichiers C++

• mlir/lib/Dialect/SparseTensor
• Bibliothèque d'assistance pour l'exécution de tenseurs creux pour le moteur d'exécution
  • mlir/lib/ExecutionEngine/SparseTensor (Runtime)

Liaisons Python et 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

• mlir/benchmark/python/benchmark_sparse.py

Tests

• mlir/test/Dialect/SparseTensor (tests FileCheck)
• mlir/test/Integration/Dialect/SparseTensor (tests de bout en bout)
• mlir/unittests/Dialect/SparseTensor (tests unitaires)
• mlir/test/CAPI/sparse_tensor.c (tests CAPI)
• mlir/test/python/dialects/sparse_tensor/dialect.py (tests de liaison Python)