From 62b7b986f2527ab31cee9a5100aae0a74a29350e Mon Sep 17 00:00:00 2001 From: getumen Date: Tue, 23 Jan 2024 15:17:30 +0900 Subject: [PATCH] fix sync method (#30) --- .devcontainer/Dockerfile | 6 +++--- rust/src/sys/bindings.rs | 2 +- src/agglomerative_clustering.cu | 2 +- src/dbscan.cu | 2 +- src/fil.cu | 2 +- src/kmeans.cu | 2 +- src/linear_regression.cu | 6 +++--- 7 files changed, 11 insertions(+), 11 deletions(-) diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile index a06fb81..55ecf61 100644 --- 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 index 988c862..77d24da 100644 --- 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 index f8a4422..5d932f8 100644 --- 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 index 98b8ca4..ee87ede 100644 --- 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 index 3903d01..88f6d56 100644 --- 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 index 2d1eed0..89ad429 100644 --- 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 index 7275cc4..73d851f 100644 --- 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; }