DEVICE/CUDA_IPC: Fix proto selection for cuda_ipc (#10943)

Author: michal-shalev

src/ucp/wireup/select.c

@@ -2435,6 +2435,18 @@ ucp_wireup_select_context_init(ucp_wireup_select_context_t *select_ctx)
     UCS_STATIC_BITMAP_RESET_ALL(&select_ctx->tl_bitmap);
 }
 
+static double ucp_wireup_device_score_func(const ucp_worker_iface_t *wiface,
+                                           const uct_md_attr_v2_t *md_attr,
+                                           const ucp_unpacked_address_t *unpacked_addr,
+                                           const ucp_address_entry_t *remote_addr,
+                                           int is_prioritized_ep, void *arg)
+{
+    ucp_wireup_dev_usage_count *dev_count = arg;
+
+    return ucp_wireup_iface_avail_bandwidth(wiface, unpacked_addr, remote_addr,
+                                            dev_count) / UCS_MBYTE;
+}
+
 /* Also ignore error mode when set to peer failure */
 static ucs_status_t
 ucp_wireup_add_device_lanes(const ucp_wireup_select_params_t *select_params,
@@ -2462,7 +2474,7 @@ ucp_wireup_add_device_lanes(const ucp_wireup_select_params_t *select_params,
     ucp_wireup_init_select_flags(&peer_rma_flags, 0, 0);
     ucp_wireup_criteria_init(&bw_info.criteria);
 
-    bw_info.criteria.calc_score = ucp_wireup_rma_bw_score_func;
+    bw_info.criteria.calc_score = ucp_wireup_device_score_func;
     ucp_wireup_init_select_flags(&bw_info.criteria.local_iface_flags,
                                  UCT_IFACE_FLAG_DEVICE_EP, 0);