Home
last modified time | relevance | path

Searched refs:isSmallAlloc (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DBufferOptimizations.cpp329 void promote(function_ref<bool(Value)> isSmallAlloc) { in promote() argument
337 if (!isSmallAlloc(alloc) || dealloc || in promote()
396 explicit PromoteBuffersToStackPass(std::function<bool(Value)> isSmallAlloc) in PromoteBuffersToStackPass() argument
397 : isSmallAlloc(std::move(isSmallAlloc)) {} in PromoteBuffersToStackPass()
400 if (isSmallAlloc == nullptr) { in initialize()
401 isSmallAlloc = [=](Value alloc) { in initialize()
412 optimizer.promote(isSmallAlloc); in runOnOperation()
416 std::function<bool(Value)> isSmallAlloc; member in __anon68c8f7d70211::PromoteBuffersToStackPass
436 std::function<bool(Value)> isSmallAlloc) { in createPromoteBuffersToStackPass() argument
437 return std::make_unique<PromoteBuffersToStackPass>(std::move(isSmallAlloc)); in createPromoteBuffersToStackPass()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/Transforms/
H A DPasses.h72 createPromoteBuffersToStackPass(std::function<bool(Value)> isSmallAlloc);