Skip to content

Commit 2a2e058

Browse files
Fixes
1 parent 4cca094 commit 2a2e058

File tree

4 files changed

+7
-5
lines changed

4 files changed

+7
-5
lines changed

lib/gc/ExecutionEngine/GPURuntime/ocl/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ get_property(GC_JIT_WRAPPER_LIBS GLOBAL PROPERTY GC_JIT_WRAPPER_LIBS)
44
gc_add_mlir_library(GcGpuOclRuntime
55
GpuOclRuntime.cpp
66
LINK_LIBS PUBLIC
7-
GcPasses
7+
GcGpuPasses
88
${GC_JIT_WRAPPER_LIBS}
99
)
1010
target_include_directories(GcGpuOclRuntime PUBLIC ${OpenCL_INCLUDE_DIRS})

lib/gc/ExecutionEngine/GPURuntime/ocl/GpuOclRuntime.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ struct OclRuntime::Exports {
163163
llvm::orc::SymbolMap map;
164164
map.reserve(6);
165165
map.try_emplace(interner(GPU_OCL_MALLOC),
166-
llvm::orc::ExecutorAddr::fromPtr(&malloc),
166+
llvm::orc::ExecutorAddr::fromPtr(&alloc),
167167
llvm::JITSymbolFlags::Exported);
168168
map.try_emplace(interner(GPU_OCL_DEALLOC),
169169
llvm::orc::ExecutorAddr::fromPtr(&dealloc),
@@ -184,7 +184,7 @@ struct OclRuntime::Exports {
184184
}
185185

186186
private:
187-
static void *maloc(const OclContext *ctx, size_t size) {
187+
static void *alloc(const OclContext *ctx, size_t size) {
188188
return gcGetOrReport(ctx->runtime.usmAllocDev(size));
189189
}
190190

lib/gc/Transforms/GPU/Pipeline.cpp

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

79+
// Temporary disabled until #344 is implemented
7980
// pm.addNestedPass<func::FuncOp>(imex::createInsertGPUAllocsPass("opencl"));
8081
pm.addPass(createGpuKernelOutliningPass());
8182
pm.addPass(createCanonicalizerPass());
Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
1-
if not config.gc_use_imex:
2-
config.unsupported = True
1+
#if not config.gc_use_imex:
2+
# GPUX is currently disabled
3+
config.unsupported = True

0 commit comments

Comments
 (0)