Cấu trúc kho lưu trữ phương ngữ Tensor thưa

Ngôn ngữ lập trình Sparse Tensor nằm trong kho lưu trữ MLIR.

Tệp tiêu đề

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

Thư viện và tệp C++

• mlir/lib/Dialect/SparseTensor
• Thư viện hỗ trợ thời gian chạy tensor thưa thớt cho công cụ thực thi
  • mlir/lib/ExecutionEngine/SparseTensor (Thời gian chạy)

Các liên kết Python và 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

Điểm chuẩn MLIR

• mlir/benchmark/python/benchmark_sparse.py

Thử nghiệm

• mlir/test/Dialect/SparseTensor (Kiểm thử FileCheck)
• mlir/test/Integration/Dialect/SparseTensor (Kiểm thử toàn diện)
• mlir/unittests/Dialect/SparseTensor (Kiểm thử đơn vị)
• mlir/test/CAPI/sparse_tensor.c (Các kiểm thử CAPI)
• mlir/test/python/dialects/sparse_tensor/dialect.py (Các kiểm thử liên kết Python)