update

minmingzhu · Jul 29, 2024 · 4cb9289 · 4cb9289
1 parent 3e55f74
commit 4cb9289
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mllib-dal/src/main/native/CorrelationImpl.cpp b/mllib-dal/src/main/native/CorrelationImpl.cpp
@@ -318,11 +318,11 @@ Java_com_intel_oap_mllib_stat_CorrelationDALImpl_cCorrelationTrainDAL(
             logger::INFO,
             "oneDAL (native): use GPU kernels with %d GPU(s) rankid %d", nGpu,
             rank);
+        jint *gpuIndices = env->GetIntArrayElements(gpuIdxArray, 0);
 
-        //        auto queue = getGPU(device, gpuIndices);
+        auto queue = getGPU(device, gpuIndices);
         auto gpu_device = sycl::device(sycl::gpu_selector_v);
         sycl::queue queue{gpu_device};
-        jint *gpuIndices = env->GetIntArrayElements(gpuIdxArray, 0);
         const char* cstr = env->GetStringUTFChars(breakdown_name, nullptr);
         std::string c_breakdown_name(cstr);
         const char *str = env->GetStringUTFChars(ip_port, 0);