static ucs_status_t uct_cma_query_tl_resources(uct_pd_h pd, uct_tl_resource_desc_t **resource_p, unsigned *num_resources_p) { uct_tl_resource_desc_t *resource; resource = ucs_calloc(1, sizeof(uct_tl_resource_desc_t), "resource desc"); if (NULL == resource) { ucs_error("Failed to allocate memory"); return UCS_ERR_NO_MEMORY; } ucs_snprintf_zero(resource->tl_name, sizeof(resource->tl_name), "%s", UCT_CMA_TL_NAME); ucs_snprintf_zero(resource->dev_name, sizeof(resource->dev_name), "%s", pd->component->name); resource->dev_type = UCT_DEVICE_TYPE_SHM; *num_resources_p = 1; *resource_p = resource; return UCS_OK; }
static ucs_status_t uct_gdr_copy_query_tl_resources(uct_md_h md, uct_tl_resource_desc_t **resource_p, unsigned *num_resources_p) { uct_tl_resource_desc_t *resource; resource = ucs_calloc(1, sizeof(uct_tl_resource_desc_t), "resource desc"); if (NULL == resource) { ucs_error("Failed to allocate memory"); return UCS_ERR_NO_MEMORY; } ucs_snprintf_zero(resource->tl_name, sizeof(resource->tl_name), "%s", UCT_GDR_COPY_TL_NAME); ucs_snprintf_zero(resource->dev_name, sizeof(resource->dev_name), "%s", UCT_CUDA_DEV_NAME); resource->dev_type = UCT_DEVICE_TYPE_ACC; *num_resources_p = 1; *resource_p = resource; return UCS_OK; }
static ucs_status_t uct_cuda_query_tl_resources(uct_pd_h pd, uct_tl_resource_desc_t **resource_p, unsigned *num_resources_p) { uct_tl_resource_desc_t *resource; resource = ucs_calloc(1, sizeof(uct_tl_resource_desc_t), "resource desc"); if (NULL == resource) { ucs_error("Failed to allocate memory"); return UCS_ERR_NO_MEMORY; } ucs_snprintf_zero(resource->tl_name, sizeof(resource->tl_name), "%s", UCT_CUDA_TL_NAME); ucs_snprintf_zero(resource->dev_name, sizeof(resource->dev_name), "%s", UCT_CUDA_DEV_NAME); resource->latency = 1; /* FIXME temp value */ resource->bandwidth = (long) (6911 * pow(1024,2)); /* FIXME temp value */ *num_resources_p = 1; *resource_p = resource; return UCS_OK; }
static ucs_status_t uct_self_query_tl_resources(uct_md_h md, uct_tl_resource_desc_t **resource_p, unsigned *num_resources_p) { uct_tl_resource_desc_t *resource = 0; ucs_trace_func("md=%p", md); resource = ucs_calloc(1, sizeof(*resource), "resource desc"); if (NULL == resource) { ucs_error("Failed to allocate memory"); return UCS_ERR_NO_MEMORY; } ucs_snprintf_zero(resource->tl_name, sizeof(resource->tl_name), "%s", UCT_SELF_NAME); ucs_snprintf_zero(resource->dev_name, sizeof(resource->dev_name), "%s", UCT_SELF_NAME); resource->dev_type = UCT_DEVICE_TYPE_SELF; *num_resources_p = 1; *resource_p = resource; return UCS_OK; }