بنية مستودع لغة Sparse Tensor

يمكن العثور على Sparse Tensor Dialect في مستودع 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 (Passes)
• mlir/python/mlir/dialects/sparse_tensor.py

MLIR Benchmark

• 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)