Home
last modified time | relevance | path

Searched defs:subgroupSize (Results 1 – 6 of 6) sorted by relevance

/llvm-project/mlir/lib/Dialect/GPU/Transforms/
H A DSubgroupReduceLowering.cpp153 createSubgroupShuffleReduction(OpBuilder & builder,Location loc,Value input,gpu::AllReduceOperation mode,unsigned subgroupSize,function_ref<Value (Value)> packFn,function_ref<Value (Value)> unpackFn) createSubgroupShuffleReduction() argument
225 unsigned subgroupSize = 0; global() member
303 unsigned subgroupSize = 0; global() member
317 populateGpuLowerSubgroupReduceToShufflePattenrs(RewritePatternSet & patterns,unsigned subgroupSize,unsigned shuffleBitwidth,PatternBenefit benefit) populateGpuLowerSubgroupReduceToShufflePattenrs() argument
[all...]
H A DAllReduceLowering.cpp285 Value subgroupSize = create<arith::ConstantIntOp>(kSubgroupSize, int32Type); in createSubgroupReduce() local
334 Value subgroupSize = create<arith::ConstantIntOp>(kSubgroupSize, int32Type); in getDivideBySubgroupSize() local
/llvm-project/mlir/test/lib/Dialect/SPIRV/
H A DTestEntryPointAbi.cpp48 Pass::Option<int> subgroupSize{ member
/llvm-project/mlir/lib/Dialect/SPIRV/IR/
H A DTargetAndABI.cpp125 std::optional<int> subgroupSize, std::optional<int> targetWidth) { in getEntryPointABIAttr()
/llvm-project/mlir/lib/Dialect/SPIRV/Transforms/
H A DLowerABIAttributesPass.cpp163 if (std::optional<int> subgroupSize = entryPointAttr.getSubgroupSize()) { lowerEntryPointABIAttr() local
/llvm-project/mlir/lib/Conversion/GPUToSPIRV/
H A DGPUToSPIRV.cpp433 unsigned subgroupSize = matchAndRewrite() local