update

minmingzhu · Jul 30, 2024 · 28b0488 · 28b0488
1 parent 121ccc1
commit 28b0488
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
@@ -333,7 +333,7 @@ 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 gpus = test_gpus()
+        auto gpus = test_gpus();
 //        auto queue = getGPU(device, gpuIndices);
 //        auto gpu_device = sycl::device(sycl::gpu_selector_v);
 //        sycl::queue queue{gpu_device};
@@ -373,7 +373,7 @@ Java_com_intel_oap_mllib_stat_CorrelationDALImpl_cCorrelationTrainDAL(
         logger::println(logger::INFO, "OneCCL (native): init took %f secs",
                         duration / 1000);
         logger::Logger::getInstance(c_breakdown_name).printLogToFile("rankID was %d, OneCCL create communicator took %f secs.", rank, duration / 1000 );
-        sycl::queue queue{gpus[gpu_indices[0]]};
+        sycl::queue queue{gpus[gpuIndices[0]]};
 
         t1 = std::chrono::high_resolution_clock::now();
         auto comm =