fix distanceLauncher invocation

rapidsai · Feb 13, 2023 · 68c5bcd · 68c5bcd
1 parent 3270026
commit 68c5bcd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cpp/test/prims/distance_base.cuh b/cpp/test/prims/distance_base.cuh
@@ -204,7 +204,8 @@ class DistanceTest : public ::testing::TestWithParam<DistanceInputs<DataType>> {
     rmm::device_uvector<char> workspace(worksize);
 
     DataType threshold = -10000.f;
-    distanceLauncher<distanceType, DataType>(x.data(),
+    distanceLauncher<distanceType, DataType>(handle,
+                                             x.data(),
                                              y.data(),
                                              dist.data(),
                                              dist2.data(),
@@ -215,7 +216,6 @@ class DistanceTest : public ::testing::TestWithParam<DistanceInputs<DataType>> {
                                              threshold,
                                              workspace.data(),
                                              worksize,
-                                             stream,
                                              isRowMajor);
   }