update

minmingzhu · Jul 29, 2024 · dafd254 · dafd254
1 parent a9fb3b1
commit dafd254
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/mllib-dal/src/main/native/CorrelationImpl.cpp b/mllib-dal/src/main/native/CorrelationImpl.cpp
@@ -319,10 +319,9 @@ Java_com_intel_oap_mllib_stat_CorrelationDALImpl_cCorrelationTrainDAL(
             "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 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, nullptr);