#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include <mpi.h>
#include <doca_ctx.h>
#include <doca_dev.h>
#include <doca_error.h>
#include <doca_log.h>
#include <doca_pe.h>
#include <doca_urom.h>
#include <worker_sandbox.h>
#include "common.h"
#include "urom_common.h"
Go to the source code of this file.
|
| | DOCA_LOG_REGISTER (UROM_PING_PONG::SAMPLE) |
| |
| static doca_error_t | oob_allgather (void *sbuf, void *rbuf, size_t msglen, void *coll_info, void **req) |
| |
| static doca_error_t | oob_allgather_test (void *req) |
| |
| static doca_error_t | oob_allgather_free (void *req) |
| |
| static void | recv_finished_cb (doca_error_t result, union doca_data cookie, union doca_data context, void *buffer, uint64_t buf_len, uint64_t sender_tag, ucs_status_t status) |
| |
| static void | send_finished_cb (doca_error_t result, union doca_data cookie, union doca_data context, ucs_status_t status) |
| |
| doca_error_t | ping_pong (struct doca_pe *pe, struct doca_urom_worker *worker, const char *msg, int my_rank, int size) |
| |
| doca_error_t | urom_ping_pong (const char *message, const char *device_name, uint32_t rank, uint32_t size) |
| |
◆ PING_RECV_CTX
| #define PING_RECV_CTX 0xdead /* Ping recv context */ |
◆ PING_TAG
| #define PING_TAG 0x1234 /* UCX data for ping tag */ |
◆ PONG_RECV_CTX
| #define PONG_RECV_CTX 0xbeef /* Pong recv context */ |
◆ PONG_TAG
| #define PONG_TAG 0x5678 /* UCX data for pong tag */ |
◆ RECV_MAX_LEN
| #define RECV_MAX_LEN 100 /* Receive message maximum size */ |
◆ pp_res_type_t
◆ DOCA_LOG_REGISTER()
| DOCA_LOG_REGISTER |
( |
UROM_PING_PONG::SAMPLE |
| ) |
|
◆ oob_allgather()
| static doca_error_t oob_allgather |
( |
void * |
sbuf, |
|
|
void * |
rbuf, |
|
|
size_t |
msglen, |
|
|
void * |
coll_info, |
|
|
void ** |
req |
|
) |
| |
|
static |
◆ oob_allgather_free()
◆ oob_allgather_test()
◆ ping_pong()
| doca_error_t ping_pong |
( |
struct doca_pe * |
pe, |
|
|
struct doca_urom_worker * |
worker, |
|
|
const char * |
msg, |
|
|
int |
my_rank, |
|
|
int |
size |
|
) |
| |
◆ recv_finished_cb()
| static void recv_finished_cb |
( |
doca_error_t |
result, |
|
|
union doca_data |
cookie, |
|
|
union doca_data |
context, |
|
|
void * |
buffer, |
|
|
uint64_t |
buf_len, |
|
|
uint64_t |
sender_tag, |
|
|
ucs_status_t |
status |
|
) |
| |
|
static |
◆ send_finished_cb()
◆ urom_ping_pong()
| doca_error_t urom_ping_pong |
( |
const char * |
message, |
|
|
const char * |
device_name, |
|
|
uint32_t |
rank, |
|
|
uint32_t |
size |
|
) |
| |