fix sync method (#30)

getumen · Jan 23, 2024 · 62b7b98 · 62b7b98
1 parent 176539c
commit 62b7b98
Show file tree

Hide file tree

Showing 7 changed files with 11 additions and 11 deletions.
diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile
@@ -6,9 +6,9 @@ USER root
 
 RUN sed -i -r 's@http://(jp\.)?archive\.ubuntu\.com/ubuntu/?@http://ftp.jaist.ac.jp/pub/Linux/ubuntu/@g' /etc/apt/sources.list
 
-ENV CPATH=/opt/conda/include:/opt/conda/include/rapids
-ENV LIBRARY_PATH=$LIBRARY_PATH:/opt/conda/lib:/opt/conda/lib/rapids
-ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/conda/lib:/opt/conda/lib/rapids
+ENV CPATH=/opt/conda/include:/opt/conda/include/rapids:/usr/local/include
+ENV LIBRARY_PATH=$LIBRARY_PATH:/opt/conda/lib:/opt/conda/lib/rapids:/usr/local/lib
+ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/conda/lib:/opt/conda/lib/rapids:/usr/local/lib
 
 RUN apt-get update \
     && apt-get install -y \

diff --git a/rust/src/sys/bindings.rs b/rust/src/sys/bindings.rs
@@ -1,4 +1,4 @@
-/* automatically generated by rust-bindgen 0.66.1 */
+/* automatically generated by rust-bindgen 0.69.2 */
 
 #![allow(non_camel_case_types, non_snake_case, non_upper_case_globals, unused)]
 

diff --git a/src/agglomerative_clustering.cu b/src/agglomerative_clustering.cu
@@ -80,7 +80,7 @@ __host__ int AgglomerativeClusteringFit(
                       d_children.size(),
                       handle_p->handle->get_stream());
 
-    handle_p->handle->get_stream().synchronize();
+    handle_p->handle->sync_stream();
 
     return 0;
 }
diff --git a/src/dbscan.cu b/src/dbscan.cu
@@ -53,7 +53,7 @@ __host__ int DbscanFit(
                       d_labels.size(),
                       handle_p->handle->get_stream());
 
-    handle_p->handle->get_stream().synchronize();
+    handle_p->handle->sync_stream();
 
     return 0;
 }
diff --git a/src/fil.cu b/src/fil.cu
@@ -209,7 +209,7 @@ __host__ int FILPredict(
                     d_preds.size(),
                     handle_p->handle->get_stream());
 
-  handle_p->handle->get_stream().synchronize();
+  handle_p->handle->sync_stream();
 
   return FIL_SUCCESS;
 }
diff --git a/src/kmeans.cu b/src/kmeans.cu
@@ -79,7 +79,7 @@ KmeansFit(
                       d_centroids.size(),
                       handle_p->handle->get_stream());
 
-    handle_p->handle->get_stream().synchronize();
+    handle_p->handle->sync_stream();
 
     return 0;
 }
diff --git a/src/linear_regression.cu b/src/linear_regression.cu
@@ -62,7 +62,7 @@ __host__ int OlsFit(
                       d_coef.size(),
                       handle_p->handle->get_stream());
 
-    handle_p->handle->get_stream().synchronize();
+    handle_p->handle->sync_stream();
 
     return 0;
 }
@@ -125,7 +125,7 @@ __host__ int RidgeFit(
                       d_coef.size(),
                       handle_p->handle->get_stream());
 
-    handle_p->handle->get_stream().synchronize();
+    handle_p->handle->sync_stream();
 
     return 0;
 }
@@ -177,7 +177,7 @@ __host__ int GemmPredict(
                       d_preds.size(),
                       handle_p->handle->get_stream());
 
-    handle_p->handle->get_stream().synchronize();
+    handle_p->handle->sync_stream();
 
     return 0;
 }