Home
last modified time | relevance | path

Searched refs:LinalgDependenceGraph (Results 1 – 8 of 8) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Linalg/Analysis/
DDependenceAnalysis.cpp74 StringRef LinalgDependenceGraph::getDependenceTypeStr(DependenceType depType) { in getDependenceTypeStr()
76 case LinalgDependenceGraph::DependenceType::RAW: in getDependenceTypeStr()
78 case LinalgDependenceGraph::DependenceType::RAR: in getDependenceTypeStr()
80 case LinalgDependenceGraph::DependenceType::WAR: in getDependenceTypeStr()
82 case LinalgDependenceGraph::DependenceType::WAW: in getDependenceTypeStr()
90 LinalgDependenceGraph
91 LinalgDependenceGraph::buildDependenceGraph(Aliases &aliases, FuncOp f) { in buildDependenceGraph()
94 return LinalgDependenceGraph(aliases, linalgOps); in buildDependenceGraph()
97 LinalgDependenceGraph::LinalgDependenceGraph(Aliases &aliases, in LinalgDependenceGraph() function in LinalgDependenceGraph
111 void LinalgDependenceGraph::addDependenceElem(DependenceType dt, in addDependenceElem()
[all …]
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/Utils/
DUtils.h34 class LinalgDependenceGraph; variable
76 bool isProducerLastWriteOfView(const LinalgDependenceGraph &graph,
83 bool isFusableInto(const LinalgDependenceGraph &graph, LinalgOp consumer,
88 SmallVector<LinalgDependenceGraph::LinalgDependenceGraphElem, 1>>;
91 const LinalgDependenceGraph &dependenceGraph);
101 const LinalgDependenceGraph &graph);
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DFusion.cpp300 bool mlir::linalg::isProducerLastWriteOfView(const LinalgDependenceGraph &graph, in isProducerLastWriteOfView()
324 bool mlir::linalg::isFusableInto(const LinalgDependenceGraph &graph, in isFusableInto()
380 static Optional<LinalgDependenceGraph::LinalgDependenceGraphElem>
382 const LinalgDependenceGraph &dependenceGraph) { in findFusableProducer()
385 LinalgDependenceGraph::DependenceType::RAW, in findFusableProducer()
386 LinalgDependenceGraph::DependenceType::WAW, in findFusableProducer()
391 LinalgDependenceGraph::LinalgDependenceGraphElem elem) { in findFusableProducer()
408 << LinalgDependenceGraph::getDependenceTypeStr(depType) in findFusableProducer()
429 const LinalgDependenceGraph &graph) { in fuseProducerOfBuffer()
430 Optional<LinalgDependenceGraph::LinalgDependenceGraphElem> fusableDependence = in fuseProducerOfBuffer()
[all …]
DTransforms.cpp154 const LinalgDependenceGraph &dependenceGraph, in LinalgBaseTileAndFusePattern()
/external/llvm-project/mlir/test/lib/Transforms/
DTestLinalgFusionTransforms.cpp39 const LinalgDependenceGraph &dependenceGraph, in fillFusionPatterns()
113 LinalgDependenceGraph dependenceGraph = in applyFusionPatterns()
114 LinalgDependenceGraph::buildDependenceGraph(alias, funcOp); in applyFusionPatterns()
144 linalg::LinalgDependenceGraph graph(aliases, linalgOps); in fuseLinalgOpsGreedily()
227 LinalgDependenceGraph dependenceGraph(aliases, linalgOps); in runOnFunction()
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/Analysis/
DDependenceAnalysis.h45 class LinalgDependenceGraph {
72 static LinalgDependenceGraph buildDependenceGraph(Aliases &aliases, FuncOp f);
73 LinalgDependenceGraph(Aliases &aliases, ArrayRef<LinalgOp> ops);
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/
DTransforms.h130 const LinalgDependenceGraph &dependenceGraph,
446 const LinalgDependenceGraph &dependenceGraph,
458 const LinalgDependenceGraph &dependenceGraph;
478 const LinalgDependenceGraph &dependenceGraph,
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_fuse_linalg.cc176 linalg::LinalgDependenceGraph graph(aliases, linalg_ops); in runOnFunction()