H A D | Sparsification.cpp | 248 static bool findSparseAnnotations(CodegenEnv &env, bool idxReducBased) { in findSparseAnnotations() argument 287 static void genBuffers(CodegenEnv &env, OpBuilder &builder) { in genBuffers() argument 337 genIndex(CodegenEnv & env,OpOperand * t) genIndex() argument 349 genSubscript(CodegenEnv & env,OpBuilder & builder,OpOperand * t,SmallVectorImpl<Value> & args) genSubscript() argument 374 genInsertionLoad(CodegenEnv & env,OpBuilder & builder,OpOperand * t) genInsertionLoad() argument 389 genInsertionLoadReduce(CodegenEnv & env,OpBuilder & builder,OpOperand * t) genInsertionLoadReduce() argument 423 genInsertionStore(CodegenEnv & env,OpBuilder & builder,OpOperand * t,Value rhs) genInsertionStore() argument 496 genTensorLoad(CodegenEnv & env,OpBuilder & builder,ExprId exp) genTensorLoad() argument 522 genTensorStore(CodegenEnv & env,OpBuilder & builder,ExprId exp,Value rhs) genTensorStore() argument 575 genInvariantValue(CodegenEnv & env,ExprId exp) genInvariantValue() argument 584 relinkBranch(CodegenEnv & env,RewriterBase & rewriter,Block * block,Value e) relinkBranch() argument 618 genExp(CodegenEnv & env,RewriterBase & rewriter,ExprId e) genExp() argument 677 genInvariants(CodegenEnv & env,OpBuilder & builder,ExprId exp,LoopId curr,bool isStart) genInvariants() argument 747 genExpand(CodegenEnv & env,OpBuilder & builder,LoopId curr,bool isStart) genExpand() argument 791 isParallelFor(CodegenEnv & env,bool isOuter,bool isSparse) isParallelFor() argument 816 shouldTryParallize(CodegenEnv & env,LoopId curr,ArrayRef<TensorLevel> tidLvls) shouldTryParallize() argument 833 genCoIteration(CodegenEnv & env,OpBuilder & builder,ArrayRef<TensorLevel> tidLvls,bool tryParallel,bool needsUniv) genCoIteration() argument 847 genLoop(CodegenEnv & env,OpBuilder & builder,LoopId curr,bool needsUniv,ArrayRef<TensorLevel> tidLvls) genLoop() argument 854 finalizeWhileOp(CodegenEnv & env,OpBuilder & builder,bool needsUniv) finalizeWhileOp() argument 894 genIf(CodegenEnv & env,OpBuilder & builder,LoopId curr,LatPointId p) genIf() argument 941 endIf(CodegenEnv & env,OpBuilder & builder,scf::IfOp ifOp,Value redInput,Value cntInput,Value insInput,Value validIns) endIf() argument 972 getAllTidLvlsInLatPoints(CodegenEnv & env,LatPointId li,LoopId curr,llvm::function_ref<void (TensorLevel,AffineExpr)> callback) getAllTidLvlsInLatPoints() argument 1083 startLoopSeq(CodegenEnv & env,OpBuilder & builder,ExprId exp,LoopId curr,LatSetId lts) startLoopSeq() argument 1115 genConstantDenseAddressFromLevel(CodegenEnv & env,OpBuilder & builder,TensorId tid,Level startLvl) genConstantDenseAddressFromLevel() argument 1145 genInitConstantDenseAddress(CodegenEnv & env,RewriterBase & rewriter) genInitConstantDenseAddress() argument 1153 translateBitsToTidLvlPairs(CodegenEnv & env,LatPointId li,LoopId curr,SmallVectorImpl<TensorLevel> & tidLvls,SmallVectorImpl<std::pair<TensorLevel,AffineExpr>> & affineTidLvls) translateBitsToTidLvlPairs() argument 1166 startLoop(CodegenEnv & env,OpBuilder & builder,LoopId curr,LatPointId li,bool needsUniv) startLoop() argument 1200 endLoop(CodegenEnv & env,RewriterBase & rewriter,Operation * loop,LatPointId li,bool needsUniv,bool isSingleCond) endLoop() argument 1221 endLoopSeq(CodegenEnv & env,OpBuilder & builder,unsigned exp,unsigned at) endLoopSeq() argument 1234 genStmt(CodegenEnv & env,RewriterBase & rewriter,ExprId exp,LoopId curr) genStmt() argument 1296 genResult(CodegenEnv & env,RewriterBase & rewriter) genResult() argument 1370 CodegenEnv env(op, options, numTensors, numLoops, maxLvlRank); matchAndRewrite() local [all...] |