llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/SparseReinterpretMap.cpp
Aart Bik 7cfac1bedd
[mlir][sparse] add boilterplate code for a new reintepret map pass (#70393)
The interesting stuff is of course still coming ;-)
2023-10-26 17:57:46 -07:00

23 lines
849 B
C++

//===- SparseReinterpretMap.cpp - reinterpret sparse tensor maps ----------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "mlir/Dialect/SparseTensor/IR/SparseTensor.h"
#include "mlir/Dialect/SparseTensor/IR/SparseTensorType.h"
#include "mlir/Dialect/SparseTensor/Transforms/Passes.h"
namespace {
// TODO:
// (1) insert the zero-cost sparse_tensor.reinterpret_map ops
// (2) rewrite linalg.generic ops traits on level crds
// (3) compute topsort, and resolve cyles with sparse_tensor.convert ops
} // namespace
void mlir::populateSparseReinterpretMap(RewritePatternSet &patterns) {}