Searched refs:CodegenEnv (Results 1 – 4 of 4) sorted by relevance
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/ |
| H A D | CodegenEnv.cpp | 44 CodegenEnv::CodegenEnv(linalg::GenericOp linop, SparsificationOptions opts, in CodegenEnv() function in CodegenEnv
52 LogicalResult CodegenEnv::initTensorExp() { in initTensorExp()
62 void CodegenEnv::startEmit(SparseEmitStrategy emitStrategy) { in startEmit()
103 std::optional<Operation *> CodegenEnv::genLoopBoundary( in genLoopBoundary()
136 bool CodegenEnv::isAdmissibleTensorExp(ExprId exp) { in isAdmissibleTensorExp()
187 Value CodegenEnv::getLoopVar(LoopId i) const { in getLoopVar()
195 void CodegenEnv::updateInsertionChain(Value chain) { in updateInsertionChain()
200 bool CodegenEnv::atExpandLevel(OpOperand *o, unsigned rank, LoopId n) const { in atExpandLevel()
205 void CodegenEnv::startExpand(Value values, Value filled, Value added, in startExpand()
214 void CodegenEnv::updateExpandCount(Value count) { in updateExpandCount()
[all …]
|
| H A D | CodegenEnv.h | 1 //===- CodegenEnv.h - Code generation environment class ---------*- C++ -*-===//
35 class CodegenEnv {
40 CodegenEnv(linalg::GenericOp linop, SparsificationOptions opts,
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | Sparsification.cpp | 13 #include "Utils/CodegenEnv.h"
248 static bool findSparseAnnotations(CodegenEnv &env, bool idxReducBased) { in findSparseAnnotations()
287 static void genBuffers(CodegenEnv &env, OpBuilder &builder) { in genBuffers()
337 static Value genIndex(CodegenEnv &env, OpOperand *t) { in genIndex()
349 static Value genSubscript(CodegenEnv &env, OpBuilder &builder, OpOperand *t, in genSubscript()
377 static Value genInsertionLoad(CodegenEnv &env, OpBuilder &builder, in genInsertionLoad()
392 static Value genInsertionLoadReduce(CodegenEnv &env, OpBuilder &builder, in genInsertionLoadReduce()
426 static void genInsertionStore(CodegenEnv &env, OpBuilder &builder, OpOperand *t, in genInsertionStore()
499 static Value genTensorLoad(CodegenEnv &env, OpBuilder &builder, ExprId exp) { in genTensorLoad()
532 static void genTensorStore(CodegenEnv in genTensorStore()
[all...] |
| H A D | CMakeLists.txt | 20 Utils/CodegenEnv.cpp
|