Skip to content

T
tic
Commit 828d0266 by Tianqi Chen Committed by GitHub
Options

[OPENCL] Fix 32bit pointer size in OpenCL runtime (#809)

parent f0f696ee
Showing with 8 additions and 3 deletions
src/runtime/opencl/opencl_module.cc
...@@ -254,9 +254,14 @@ PackedFunc OpenCLModuleNode::GetFunction( ...@@ -254,9 +254,14 @@ PackedFunc OpenCLModuleNode::GetFunction(
for (size_t i = 0; i < info.arg_types.size(); ++i) { for (size_t i = 0; i < info.arg_types.size(); ++i) {
TVMType t = info.arg_types[i]; TVMType t = info.arg_types[i];
CHECK_EQ(t.lanes, 1U); CHECK_EQ(t.lanes, 1U);
uint32_t bits = t.bits; if (t.code == kHandle) {
CHECK_EQ(bits % 8, 0U); // specially store pointer type size in OpenCL driver
arg_size[i] = bits / 8; arg_size[i] = sizeof(void*);
} else {
uint32_t bits = t.bits;
CHECK_EQ(bits % 8, 0U);
arg_size[i] = bits / 8;
}
} }
// initialize the wrapped func. // initialize the wrapped func.
f.Init(this, sptr_to_self, kid_map_.at(name), f.Init(this, sptr_to_self, kid_map_.at(name),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment