稀疏张量方言代码库结构

稀疏张量方言位于 MLIR 代码库中。

头文件

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

库和 C++ 文件

• mlir/lib/Dialect/SparseTensor
• 执行引擎的稀疏张量运行时支持库
  • mlir/lib/ExecutionEngine/SparseTensor（运行时）

Python 绑定和 CAPI

• mlir/include/mlir-c/Dialect/SparseTensor.h
• mlir/lib/Bindings/Python/DialectSparseTensor.cpp（通过）
• mlir/lib/CAPI/Dialect/SparseTensor.cpp（通过）
• mlir/python/mlir/dialects/sparse_tensor.py

MLIR 基准

• mlir/benchmark/python/benchmark_sparse.py

测试

• mlir/test/Dialect/SparseTensor（FileCheck 测试）
• mlir/test/Integration/Dialect/SparseTensor（端到端测试）
• mlir/unittests/Dialect/SparseTensor（单元测试）
• mlir/test/CAPI/sparse_tensor.c（CAPI 测试）
• mlir/test/python/dialects/sparse_tensor/dialect.py（Python 绑定测试）