[ML-125][GPU] Update for Kmeans and PCA

oap-project · Sep 22, 2021 · c7fbb98 · c7fbb98
1 parent 7803d8c
commit c7fbb98
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/mllib-dal/src/assembly/assembly-cpu-gpu.xml b/mllib-dal/src/assembly/assembly-cpu-gpu.xml
@@ -43,7 +43,7 @@
         <file>
             <source>${env.TBBROOT}/lib/intel64/gcc4.8/${tbb.lib}</source>
             <outputDirectory>lib</outputDirectory>
-            <destName>libtbb.so.2</destName>
+            <destName>libtbb.so.12</destName>
         </file>
         <file>
             <source>${env.TBBROOT}/lib/intel64/gcc4.8/${tbb.malloc.lib}</source>

diff --git a/mllib-dal/src/main/native/GPU.cpp b/mllib-dal/src/main/native/GPU.cpp
@@ -57,6 +57,8 @@ sycl::device getAssignedGPU(ccl::communicator &comm, int size, int rankId,
 
     auto gpu_selected = gpu_indices[local_rank % n_gpu];
     std::cout << "GPU selected for current rank: " << gpu_selected << std::endl;
+
+    // In case gpu_selected index is larger than number of GPU SYCL devices
     auto rank_gpu = gpus[gpu_selected % gpus.size()];
 
     return rank_gpu;

diff --git a/mllib-dal/src/main/native/Makefile b/mllib-dal/src/main/native/Makefile
@@ -51,7 +51,7 @@ LIBS_COMMON := -L$(CCL_ROOT)/lib -lccl \
 ifeq ($(PLATFORM_PROFILE),CPU_ONLY_PROFILE)
   LIBS := $(LIBS_COMMON)
 else ifeq ($(PLATFORM_PROFILE),CPU_GPU_PROFILE)
-  LIBS := "$(LIBS_COMMON) -l:libonedal_sycl.a"
+  LIBS := $(LIBS_COMMON) -l:libonedal_sycl.a
 endif
 
 CPP_SRCS += \