희소 텐서 다이얼렉트 저장소 구조

스파스 텐서 다이얼렉트는 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 바인딩 테스트)