update

minmingzhu · Jul 29, 2024 · 965036a · 965036a
1 parent 019892f
commit 965036a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mllib-dal/src/main/native/CorrelationImpl.cpp b/mllib-dal/src/main/native/CorrelationImpl.cpp
@@ -320,9 +320,9 @@ Java_com_intel_oap_mllib_stat_CorrelationDALImpl_cCorrelationTrainDAL(
             rank);
         jint *gpuIndices = env->GetIntArrayElements(gpuIdxArray, 0);
 
-        getGPU(device, gpuIndices);
-        auto gpu_device = sycl::device(sycl::gpu_selector_v);
-        sycl::queue queue{gpu_device};
+        auto queue = getGPU(device, gpuIndices);
+//        auto gpu_device = sycl::device(sycl::gpu_selector_v);
+//        sycl::queue queue{gpu_device};
         const char* cstr = env->GetStringUTFChars(breakdown_name, nullptr);
         std::string c_breakdown_name(cstr);
         const char *str = env->GetStringUTFChars(ip_port, 0);