Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAllReduceLowering.cpp69 Value dimX = getDimOp<gpu::BlockDimOp>(gpu::Dimension::x); in rewrite()
70 Value dimY = getDimOp<gpu::BlockDimOp>(gpu::Dimension::y); in rewrite()
71 Value dimZ = getDimOp<gpu::BlockDimOp>(gpu::Dimension::z); in rewrite()
72 Value tidX = getDimOp<gpu::ThreadIdOp>(gpu::Dimension::x); in rewrite()
73 Value tidY = getDimOp<gpu::ThreadIdOp>(gpu::Dimension::y); in rewrite()
74 Value tidZ = getDimOp<gpu::ThreadIdOp>(gpu::Dimension::z); in rewrite()
153 Value getDimOp(gpu::Dimension dimension) { in getDimOp() function