Разреженный тензорный диалект находится в репозитории MLIR .
Заголовочные файлы
Библиотека и файлы C++
- mlir/lib/Диалект/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)