[RUNTIME][OPENCL] Bugfix

Internal regression bug to fix compilation issue
apache · Jan 4, 2025 · 5897ca7 · 5897ca7
1 parent 567eeed
commit 5897ca7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/runtime/opencl/opencl_device_api.cc b/src/runtime/opencl/opencl_device_api.cc
@@ -286,7 +286,7 @@ void OpenCLWorkspace::SetNativePtr(const tvm::runtime::NDArray& narr, void* host
 #ifdef USE_OPENCL_EXTN_QCOM
     Device dev = narr.operator->()->device;
     cl_device_id device_id = GetCLDeviceID(dev.device_id);
-    auto platform = device_info[device_id].platform_id;
+    auto platform = device_to_platform[device_id];
 
     OPENCL_CALL(clFinish(this->GetQueue(dev)));
     if (desc->host_ptr) {
@@ -313,7 +313,7 @@ void OpenCLWorkspace::SetNativePtr(const tvm::runtime::NDArray& narr, void* host
 void OpenCLWorkspace::SetPerfHint(Device dev, cl_uint perf_hint) {
 #ifdef CL_CONTEXT_PERF_HINT_QCOM
   cl_device_id device_id = GetCLDeviceID(dev.device_id);
-  auto platform = device_info[device_id].platform_id;
+  auto platform = device_to_platform[device_id];
   OPENCL_CALL(clSetPerfHintQCOM(this->contexts[platform], perf_hint));
 #endif
 }