스파스 텐서 다이얼렉트는 MLIR 저장소에 있습니다.
헤더 파일
라이브러리 및 C++ 파일
- mlir/lib/Dialect/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/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 바인딩 테스트)