Skip to content

Commit 3906c28

Browse files
committed
Add an option for insert gpu allocs to treat allocations as gpu-native
1 parent cad8a29 commit 3906c28

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

cmake/imex-version.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
6c2e414a953b9a118bce6adac21cf9d42630e674
1+
20f2eef4f6c10fcbd68d358591c8b3ef4d1b97d2

lib/gc/Transforms/GPU/Pipeline.cpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,10 @@ void populateGPUPipeline(mlir::OpPassManager &pm) {
7676
pm.addNestedPass<func::FuncOp>(createGpuMapParallelLoopsPass());
7777
pm.addNestedPass<func::FuncOp>(createParallelLoopToGpuPass());
7878

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));
8083
pm.addPass(createGpuKernelOutliningPass());
8184
pm.addPass(createCanonicalizerPass());
8285
pm.addPass(imex::createSetSPIRVCapabilitiesPass());

0 commit comments

Comments
 (0)