Searched refs:annotateFn (Results 1 – 8 of 8) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | LoopPipelining.cpp | 44 PipeliningOption::AnnotationlFnType annotateFn = nullptr; member
136 annotateFn = options.annotateFn; in initializeLoopInfo()
161 if (annotateFn) in emitPrologue()
162 annotateFn(newOp, PipeliningOption::PipelinerPart::Prologue, i); in emitPrologue()
332 if (annotateFn) in createKernel()
333 annotateFn(newOp, PipeliningOption::PipelinerPart::Kernel, 0); in createKernel()
400 if (annotateFn) in emitEpilogue()
401 annotateFn(newOp, PipeliningOption::PipelinerPart::Epilogue, i - 1); in emitEpilogue()
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/SCF/ |
| H A D | TestLoopUnrolling.cpp | 61 auto annotateFn = [this](unsigned i, Operation *op, OpBuilder b) { in runOnOperation() local
67 (void)loopUnrollByFactor(loop, unrollFactor, annotateFn); in runOnOperation()
|
| H A D | TestSCFUtils.cpp | 200 options.annotateFn = annotate; in runOnOperation()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/ |
| H A D | Utils.cpp | 368 function_ref<void(unsigned, Operation *, OpBuilder)> annotateFn, in generateUnrolledLoop() argument
374 if (!annotateFn) in generateUnrolledLoop()
375 annotateFn = [](unsigned, Operation *, OpBuilder) {}; in generateUnrolledLoop()
400 annotateFn(i, clonedOp, builder); in generateUnrolledLoop()
411 annotateFn(0, &*it, builder); in generateUnrolledLoop()
420 function_ref<void(unsigned, Operation *, OpBuilder)> annotateFn) { in loopUnrollByFactor() argument
532 annotateFn, iterArgs, yieldedValues); in loopUnrollByFactor()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SCF/Transforms/ |
| H A D | Transforms.h | 143 AnnotationlFnType annotateFn = nullptr; member
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SCF/Utils/ |
| H A D | Utils.h | 137 function_ref<void(unsigned, Operation *, OpBuilder)> annotateFn = nullptr);
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/ |
| H A D | LoopUtils.h | 50 function_ref<void(unsigned, Operation *, OpBuilder)> annotateFn = nullptr);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopUtils.cpp | 1005 function_ref<void(unsigned, Operation *, OpBuilder)> annotateFn, in generateUnrolledLoop() argument
1011 if (!annotateFn) in generateUnrolledLoop()
1012 annotateFn = [](unsigned, Operation *, OpBuilder) {}; in generateUnrolledLoop()
1037 annotateFn(i, clonedOp, builder); in generateUnrolledLoop()
1048 annotateFn(0, &*it, builder); in generateUnrolledLoop()
1093 function_ref<void(unsigned, Operation *, OpBuilder)> annotateFn) { in loopUnrollByFactor() argument
1141 /*annotateFn=*/annotateFn, in loopUnrollByFactor()
|