diff --git a/mace/kernels/opencl/depth_to_space_opencl.cc b/mace/kernels/opencl/depth_to_space_opencl.cc index c22a6e4c258195b13739e51a8a789de8cba12474..74e66467912441e121f9664b29b702b6164a0b5b 100644 --- a/mace/kernels/opencl/depth_to_space_opencl.cc +++ b/mace/kernels/opencl/depth_to_space_opencl.cc @@ -53,13 +53,14 @@ void DepthToSpaceOpFunctor::operator()( kernel_name_ss << "-D" << kernel_name << "=" << obfuscated_kernel_name; built_options.emplace(kernel_name_ss.str()); auto dt = DataTypeToEnum::value; - built_options.emplace("-DDATA_TYPE=" + DtToUpstreamCLDt(dt)); - built_options.emplace("-DCMD_DATA_TYPE=" + DtToUpstreamCLCMDDt(dt)); + built_options.emplace("-DDATA_TYPE=" + DtToCLDt(dt)); + built_options.emplace("-DCMD_DATA_TYPE=" + DtToCLCMDDt(dt)); if (runtime->IsNonUniformWorkgroupsSupported()) { built_options.emplace("-DNON_UNIFORM_WORK_GROUP"); } kernel_ = - runtime->BuildKernel("depth_to_space", kernel_name, built_options); + runtime->BuildKernel("depth_to_space", + obfuscated_kernel_name, built_options); kwg_size_ = static_cast(runtime->GetKernelMaxWorkGroupSize(kernel_)); diff --git a/mace/kernels/opencl/space_to_batch_opencl.cc b/mace/kernels/opencl/space_to_batch_opencl.cc index 31b5013b737335c40255d9d4163e1d2fb8572d68..b4ae998a56f34cf3f23266e549f7346360ae4113 100644 --- a/mace/kernels/opencl/space_to_batch_opencl.cc +++ b/mace/kernels/opencl/space_to_batch_opencl.cc @@ -51,7 +51,8 @@ void SpaceToBatchFunctor::operator()( built_options.emplace("-DNON_UNIFORM_WORK_GROUP"); } kernel_ = - runtime->BuildKernel("space_to_batch", kernel_name, built_options); + runtime->BuildKernel("space_to_batch", + obfuscated_kernel_name, built_options); kwg_size_ = static_cast(runtime->GetKernelMaxWorkGroupSize(kernel_));