File tree Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Original file line number Diff line number Diff line change 1
- 6c2e414a953b9a118bce6adac21cf9d42630e674
1
+ 20f2eef4f6c10fcbd68d358591c8b3ef4d1b97d2
Original file line number Diff line number Diff line change @@ -76,7 +76,10 @@ void populateGPUPipeline(mlir::OpPassManager &pm) {
76
76
pm.addNestedPass <func::FuncOp>(createGpuMapParallelLoopsPass ());
77
77
pm.addNestedPass <func::FuncOp>(createParallelLoopToGpuPass ());
78
78
79
- pm.addNestedPass <func::FuncOp>(imex::createInsertGPUAllocsPass (" opencl" ));
79
+ imex::InsertGPUAllocsOptions insertGPUAllocsOption{
80
+ /* clientAPI*/ " opencl" , /* inRegions*/ false , /* isUsmArgs*/ true };
81
+ pm.addNestedPass <func::FuncOp>(
82
+ imex::createInsertGPUAllocsPass (insertGPUAllocsOption));
80
83
pm.addPass (createGpuKernelOutliningPass ());
81
84
pm.addPass (createCanonicalizerPass ());
82
85
pm.addPass (imex::createSetSPIRVCapabilitiesPass ());
You can’t perform that action at this time.
0 commit comments