Searched refs:int32Type (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/ |
| H A D | AllReduceLowering.cpp | 75 Value tmp1 = create<arith::MulIOp>(int32Type, tidZ, dimY); in rewrite()
76 Value tmp2 = create<arith::AddIOp>(int32Type, tmp1, tidY); in rewrite()
77 Value tmp3 = create<arith::MulIOp>(int32Type, tmp2, dimX); in rewrite()
78 Value tmp4 = create<arith::MulIOp>(int32Type, dimX, dimY); in rewrite()
79 Value invocationIdx = create<arith::AddIOp>(int32Type, tmp3, tidX); in rewrite()
80 Value workgroupSize = create<arith::MulIOp>(int32Type, tmp4, dimZ); in rewrite()
84 create<arith::ConstantIntOp>(kSubgroupSize - 1, int32Type); in rewrite()
120 create<arith::AddIOp>(int32Type, workgroupSize, subgroupMask); in rewrite()
155 return create<arith::IndexCastOp>(int32Type, dim); in getDimOp()
375 return create<arith::DivSIOp>(int32Type, value, subgroupSize); in getDivideBySubgroupSize()
[all …]
|
| /llvm-project-15.0.7/mlir/unittests/Dialect/SPIRV/ |
| H A D | DeserializationTest.cpp | 186 auto int32Type = addIntType(32); in TEST_F() local
187 auto structType = addStructType({int32Type, int32Type}); in TEST_F()
207 auto int32Type = addIntType(32); in TEST_F() local
209 auto structType = addStructType({int32Type, int64Type}); in TEST_F()
225 auto int32Type = addIntType(32); in TEST_F() local
226 auto structType = addStructType({int32Type}); in TEST_F()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUToNVVM/ |
| H A D | LowerGpuOpsToNVVMOps.cpp | 87 auto int32Type = IntegerType::get(rewriter.getContext(), 32); in matchAndRewrite() local
93 loc, int32Type, rewriter.getI32IntegerAttr(1)); in matchAndRewrite()
95 loc, int32Type, rewriter.getI32IntegerAttr(-1)); in matchAndRewrite()
97 loc, int32Type, rewriter.getI32IntegerAttr(32)); in matchAndRewrite()
99 loc, int32Type, thirtyTwo, adaptor.width()); in matchAndRewrite()
101 Value activeMask = rewriter.create<LLVM::LShrOp>(loc, int32Type, minusOne, in matchAndRewrite()
110 rewriter.create<LLVM::SubOp>(loc, int32Type, adaptor.width(), one); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/LLVMCommon/ |
| H A D | MemRefBuilder.cpp | 470 Type int32Type = typeConverter.convertType(builder.getI32Type()); in sizeBasePtr() local
473 Value three = builder.create<LLVM::ConstantOp>(loc, int32Type, in sizeBasePtr()
|