update

minmingzhu · Jul 18, 2024 · 14ff68a · 14ff68a
1 parent 70b42da
commit 14ff68a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/mllib-dal/src/main/native/GPU.h b/mllib-dal/src/main/native/GPU.h
@@ -11,3 +11,5 @@ sycl::queue getAssignedGPU(const ComputeDevice device, ccl::communicator &comm,
                            int size, int rankId, jint *gpu_indices, int n_gpu);
 
 sycl::queue getQueue(const ComputeDevice device);
+
+sycl::queue getGPU(const ComputeDevice device,jint *gpu_indices);
diff --git a/mllib-dal/src/main/native/OneCCL.cpp b/mllib-dal/src/main/native/OneCCL.cpp
@@ -153,9 +153,9 @@ JNIEXPORT jint JNICALL Java_com_intel_oap_mllib_OneCCL_00024_c_1init(
 //    rank_id = getComm().rank();
 //    comm_size = getComm().size();
 //
-//    jclass cls = env->GetObjectClass(param);
-//    jfieldID fid_comm_size = env->GetFieldID(cls, "commSize", "J");
-//    jfieldID fid_rank_id = env->GetFieldID(cls, "rankId", "J");
+    jclass cls = env->GetObjectClass(param);
+    jfieldID fid_comm_size = env->GetFieldID(cls, "commSize", "J");
+    jfieldID fid_rank_id = env->GetFieldID(cls, "rankId", "J");
 
     env->SetLongField(param, fid_comm_size, size);
     env->SetLongField(param, fid_rank_id, rank);
Original file line number	Diff line number	Diff line change
Expand Up		@@ -11,3 +11,5 @@ sycl::queue getAssignedGPU(const ComputeDevice device, ccl::communicator &comm,
		int size, int rankId, jint *gpu_indices, int n_gpu);

		sycl::queue getQueue(const ComputeDevice device);

		sycl::queue getGPU(const ComputeDevice device,jint *gpu_indices);