โครงสร้างที่เก็บภาษา Sparse Tensor

Sparse Tensor Dialect อยู่ในที่เก็บ MLIR

ไฟล์ส่วนหัว

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

ไลบรารีและไฟล์ C++

• mlir/lib/Dialect/SparseTensor
• ไลบรารีการสนับสนุนรันไทม์ของ Sparse Tensor สำหรับเครื่องมือการดำเนินการ
  • mlir/lib/ExecutionEngine/SparseTensor (Runtime)

การเชื่อมโยง Python และ CAPI

• mlir/include/mlir-c/Dialect/SparseTensor.h
• mlir/lib/Bindings/Python/DialectSparseTensor.cpp (Passes)
• 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)