Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/test/lib/Dialect/SPIRV/
H A DTestEntryPointAbi.cpp42 Pass::ListOption<int32_t> workgroupSize{ member
58 SmallVector<int32_t, 3> workgroupSizeVec(workgroupSize.begin(), in runOnOperation()
59 workgroupSize.end()); in runOnOperation()
/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAllReduceLowering.cpp80 Value workgroupSize = create<arith::MulIOp>(int32Type, tmp4, dimZ); in rewrite() local
96 create<arith::SubIOp>(workgroupSize, numThreadsWithSmallerSubgroupId); in rewrite()
120 create<arith::AddIOp>(int32Type, workgroupSize, subgroupMask); in rewrite()