@@ -87,7 +87,7 @@ struct LaunchRewritePattern : public OpRewritePattern<mhal::LaunchOp> {
8787
8888 Value makeWait (OpBuilder b, Location loc, ArrayRef<Value> deps = {}) const {
8989 auto tokenType = b.getType <gpu::AsyncTokenType>();
90- return b. create < gpu::WaitOp>( loc, tokenType, deps).getAsyncToken ();
90+ return gpu::WaitOp::create (b, loc, tokenType, deps).getAsyncToken ();
9191 }
9292
9393 template <typename T> bool isOnDevice (const T &oprUsers) const {
@@ -121,17 +121,17 @@ struct LaunchRewritePattern : public OpRewritePattern<mhal::LaunchOp> {
121121
122122 Value allocWait = makeWait (bAlloc, loc);
123123 Type gpuMemType = opr.getType ();
124- auto dst = bAlloc. create <gpu::AllocOp>(loc, gpuMemType, tokenType,
125- ValueRange{allocWait}, ValueRange{} ,
126- ValueRange{});
124+ auto dst =
125+ gpu::AllocOp::create (bAlloc, loc, gpuMemType, tokenType ,
126+ ValueRange{allocWait}, ValueRange{}, ValueRange{});
127127 Value dstMem = dst.getResult (0 );
128128 Value dstToken = dst.getResult (1 );
129129
130130 auto makeCopy = [&]() {
131131 // always copy to device, even if it's read_access only
132132 // this way we initialize with whatever was provided by the user
133- auto memcpyToken = b. create < gpu::MemcpyOp> (
134- loc, tokenType, ValueRange{dstToken}, dstMem, opr);
133+ auto memcpyToken = gpu::MemcpyOp::create (
134+ b, loc, tokenType, ValueRange{dstToken}, dstMem, opr);
135135 dstToken = memcpyToken.getResult (0 );
136136 if (writeAccess) {
137137 // copy from device
@@ -191,8 +191,8 @@ struct LaunchRewritePattern : public OpRewritePattern<mhal::LaunchOp> {
191191 auto binaryOp = module .lookupSymbol <gpu::BinaryOp>(binaryName);
192192 if (!binaryOp) {
193193 OpBuilder b (ctx);
194- binaryOp = b. create < gpu::BinaryOp>( floc, binaryName, nullptr ,
195- ArrayRef<Attribute>({binary}));
194+ binaryOp = gpu::BinaryOp::create (b, floc, binaryName, nullptr ,
195+ ArrayRef<Attribute>({binary}));
196196
197197 SymbolTable symbolTable (module );
198198 symbolTable.insert (binaryOp);
@@ -250,8 +250,8 @@ struct LaunchRewritePattern : public OpRewritePattern<mhal::LaunchOp> {
250250 }
251251
252252 // Make gpu.launch_func
253- auto gpuLaunchOp = rw. create < gpu::LaunchFuncOp> (
254- loc,
253+ auto gpuLaunchOp = gpu::LaunchFuncOp::create (
254+ rw, loc,
255255 SymbolRefAttr::get (getContext (), binaryName,
256256 {FlatSymbolRefAttr::get (getContext (), funcName)}),
257257 gpu::KernelDim3{gridSizeIdx, oneIdx, oneIdx},
@@ -266,8 +266,8 @@ struct LaunchRewritePattern : public OpRewritePattern<mhal::LaunchOp> {
266266 auto dst = operands[diff + pair.index ()];
267267 if (gpuMem.getDefiningOp <memref::AllocOp>())
268268 std::swap (gpuMem, dst);
269- auto memcpy = rw. create < gpu::MemcpyOp>( loc, tokenType,
270- ValueRange{token}, dst, gpuMem);
269+ auto memcpy = gpu::MemcpyOp::create (rw, loc, tokenType,
270+ ValueRange{token}, dst, gpuMem);
271271 tokens.push_back (memcpy.getResult (0 ));
272272 }
273273 }
@@ -304,7 +304,7 @@ struct AwaitRewritePattern : public OpRewritePattern<mhal::AwaitOp> {
304304 if (input.getType () == tokenType) {
305305 // mhal.await with token type should never have a result type
306306 assert (op.getResultType () == std::nullopt );
307- rw. create < gpu::WaitOp>( op.getLoc (), Type (), input);
307+ gpu::WaitOp::create (rw, op.getLoc (), Type (), input);
308308 rw.eraseOp (op);
309309 return success ();
310310 }
0 commit comments