fix compile error

rapidsai · Oct 6, 2020 · 900fd11 · 900fd11
1 parent e0e696a
commit 900fd11
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/cpp/include/patterns/update_frontier_v_push_if_out_nbr.cuh b/cpp/include/patterns/update_frontier_v_push_if_out_nbr.cuh
@@ -624,12 +624,13 @@ void update_frontier_v_push_if_out_nbr(
           handle.get_stream()));  // to ensure data to be sent are ready (FIXME: this can be removed
                                   // if we use ncclSend in raft::comms)
 
-        device_isend(comm,
-                     detail::iter_to_raw_ptr(buffer_key_first + tx_offsets[i]),
-                     static_cast<size_t>(tx_counts[i]),
-                     comm_dst_rank,
-                     int{0} /* tag */,
-                     buffer_requests.data() + i * (1 + tuple_size));
+        device_isend<decltype(buffer_key_first), decltype(buffer_key_first)>(
+          comm,
+          buffer_key_first + tx_offsets[i],
+          static_cast<size_t>(tx_counts[i]),
+          comm_dst_rank,
+          int{0} /* tag */,
+          buffer_requests.data() + i * (1 + tuple_size));
         device_isend<decltype(buffer_payload_first), decltype(buffer_payload_first)>(
           comm,
           buffer_payload_first + tx_offsets[i],
@@ -659,9 +660,9 @@ void update_frontier_v_push_if_out_nbr(
                   buffer_requests.data() + (tx_counts.size() + i + 1) * (1 + tuple_size),
                   std::numeric_limits<raft::comms::request_t>::max());
       } else {
-        device_irecv(
+        device_irecv<decltype(buffer_key_first), decltype(buffer_key_first)>(
           comm,
-          detail::iter_to_raw_ptr(buffer_key_first + num_buffer_elements + rx_offsets[i]),
+          buffer_key_first + num_buffer_elements + rx_offsets[i],
           static_cast<size_t>(rx_counts[i]),
           comm_src_rank,
           int{0} /* tag */,

diff --git a/cpp/include/utilities/comm_utils.cuh b/cpp/include/utilities/comm_utils.cuh
@@ -829,4 +829,4 @@ auto get_comm_buffer_begin(BufferType& buffer)
 }
 
 }  // namespace experimental
-}  // namespace cugraph
+}  // namespace cugraph