mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2025-01-06 02:48:57 +01:00
23 lines
429 B
C
23 lines
429 B
C
|
#pragma once
|
||
|
|
|
||
|
|
struct ggml_context;
|
||
|
|
struct ggml_tensor;
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
extern "C" {
|
||
|
|
#endif
|
||
|
|
|
||
|
|
struct ggml_tensor * ggml_mpi_send_tensor(
|
||
|
|
struct ggml_context * ctx,
|
||
|
|
struct ggml_tensor *src,
|
||
|
|
int dst_rank);
|
||
|
|
struct ggml_tensor * ggml_mpi_recv_tensor(
|
||
|
|
struct ggml_context * ctx,
|
||
|
|
struct ggml_tensor *parent,
|
||
|
|
struct ggml_tensor *dst,
|
||
|
|
int src_rank);
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
}
|
||
|
|
#endif
|