shuffile_util.h

Go to the documentation of this file.

#ifndef SHUFFILE_UTIL_H
#define SHUFFILE_UTIL_H
 
#include "mpi.h"
#include "kvtree.h"
 
#define SHUFFILE_SUCCESS (0)
#define SHUFFILE_FAILURE (1)
 
#define SHUFFILE_VERSION "0.1.0"
 
#define SHUFFILE_MAX_FILENAME (1024)
 
extern int shuffile_debug;
extern int shuffile_rank;
extern char* shuffile_hostname;
 
extern int shuffile_mpi_buf_size;
extern size_t shuffile_page_size;
 
void shuffile_err(const char *fmt, ...);
 
void shuffile_warn(const char *fmt, ...);
 
void shuffile_dbg(int level, const char *fmt, ...);
 
void shuffile_abort(int rc, const char *fmt, ...);
 
#define SHUFFILE_MALLOC(X) shuffile_malloc(X, __FILE__, __LINE__);
void* shuffile_malloc(size_t size, const char* file, int line);
 
void shuffile_free(void* ptr);
 
void* shuffile_align_malloc(size_t size, size_t align);
 
void shuffile_align_free(void* buf);
 
int shuffile_str_sendrecv(
  const char* send_str, int send_rank,
  char** recv_str, int recv_rank,
  MPI_Comm comm
);
 
int shuffile_swap_file_names(
  const char* file_send, int rank_send,
        char* file_recv, size_t size_recv, int rank_recv,
  const char* dir_recv, MPI_Comm comm
);
 
int shuffile_alltrue(int flag, MPI_Comm comm);
 
#include "kvtree.h"
 
#define COPY_FILES (0)
#define MOVE_FILES (1)
 
int shuffile_swap_files(
  int swap_type,
  const char* file_send, kvtree* meta_send, int rank_send,
  const char* file_recv, kvtree* meta_recv, int rank_recv,
  MPI_Comm comm
);
 
#endif