util_quda.h

Go to the documentation of this file.

#ifndef _UTIL_QUDA_H
#define _UTIL_QUDA_H
 
#include <stdio.h>
#include <stdlib.h>
#include <enum_quda.h>
#include <comm_quda.h>
#include <tune_key.h>
 
 
QudaTune getTuning();
 
QudaVerbosity getVerbosity();
char *getOutputPrefix();
FILE *getOutputFile();
 
void setVerbosity(QudaVerbosity verbosity);
void setOutputPrefix(const char *prefix);
void setOutputFile(FILE *outfile);
 
void pushVerbosity(QudaVerbosity verbosity);
 
void popVerbosity();
 
void pushOutputPrefix(const char *prefix);
 
void popOutputPrefix();
 
bool getRankVerbosity();
 
char *getPrintBuffer();
 
char* getOmpThreadStr();
 
namespace quda {
  // forward declaration
  void saveTuneCache(bool error);
}
 
#define zeroThread (threadIdx.x + blockDim.x*blockIdx.x==0 &&           \
                    threadIdx.y + blockDim.y*blockIdx.y==0 &&           \
                    threadIdx.z + blockDim.z*blockIdx.z==0)
 
#define printfZero(...) do {                                            \
    if (zeroThread) printf(__VA_ARGS__);                                \
  } while (0)
 
#ifdef MULTI_GPU
 
#define printfQuda(...) do {                           \
  sprintf(getPrintBuffer(), __VA_ARGS__);              \
  if (getRankVerbosity()) {                            \
    fprintf(getOutputFile(), "%s", getOutputPrefix()); \
    fprintf(getOutputFile(), "%s", getPrintBuffer());  \
    fflush(getOutputFile());                           \
  }                                                    \
} while (0)
 
#define errorQuda(...)                                                                                                 \
  do {                                                                                                                 \
    fprintf(getOutputFile(), "%sERROR: ", getOutputPrefix());                                                          \
    fprintf(getOutputFile(), __VA_ARGS__);                                                                             \
    fprintf(getOutputFile(), " (rank %d, host %s, " __FILE__ ":%d in %s())\n", comm_rank_global(), comm_hostname(),    \
            __LINE__, __func__);                                                                                       \
    fprintf(getOutputFile(), "%s       last kernel called was (name=%s,volume=%s,aux=%s)\n", getOutputPrefix(),        \
            getLastTuneKey().name, getLastTuneKey().volume, getLastTuneKey().aux);                                     \
    fflush(getOutputFile());                                                                                           \
    quda::saveTuneCache(true);                                                                                         \
    comm_abort(1);                                                                                                     \
  } while (0)
 
#define warningQuda(...) do {                                   \
  if (getVerbosity() > QUDA_SILENT) {                           \
    sprintf(getPrintBuffer(), __VA_ARGS__);                     \
    if (getRankVerbosity()) {                                           \
      fprintf(getOutputFile(), "%sWARNING: ", getOutputPrefix());       \
      fprintf(getOutputFile(), "%s", getPrintBuffer());                 \
      fprintf(getOutputFile(), "\n");                                   \
      fflush(getOutputFile());                                          \
    }                                                                   \
  }                                                                     \
} while (0)
 
#else
 
#define printfQuda(...) do {                         \
  fprintf(getOutputFile(), "%s", getOutputPrefix()); \
  fprintf(getOutputFile(), __VA_ARGS__);             \
  fflush(getOutputFile());                           \
} while (0)
 
#define errorQuda(...) do {                                                  \
  fprintf(getOutputFile(), "%sERROR: ", getOutputPrefix());                  \
  fprintf(getOutputFile(), __VA_ARGS__);                                     \
  fprintf(getOutputFile(), " (" __FILE__ ":%d in %s())\n",                   \
          __LINE__, __func__);                                               \
  fprintf(getOutputFile(), "%s       last kernel called was (name=%s,volume=%s,aux=%s)\n", \
          getOutputPrefix(), getLastTuneKey().name,                          \
          getLastTuneKey().volume, getLastTuneKey().aux);                    \
  quda::saveTuneCache(true);                                            \
  comm_abort(1);                                                             \
} while (0)
 
#define warningQuda(...) do {                                 \
  if (getVerbosity() > QUDA_SILENT) {                         \
    fprintf(getOutputFile(), "%sWARNING: ", getOutputPrefix()); \
    fprintf(getOutputFile(), __VA_ARGS__);                      \
    fprintf(getOutputFile(), "\n");                             \
    fflush(getOutputFile());                                    \
  }                                                             \
} while (0)
 
#endif // MULTI_GPU
 
#define checkCudaErrorNoSync()                                                                                         \
  do {                                                                                                                 \
    cudaError_t error = cudaGetLastError();                                                                            \
    if (error != cudaSuccess) errorQuda("(CUDA) %s", cudaGetErrorString(error));                                       \
  } while (0)
 
#ifdef HOST_DEBUG
 
#define checkCudaError() do {  \
  cudaDeviceSynchronize();     \
  checkCudaErrorNoSync();      \
} while (0)
 
#else
 
#define checkCudaError() checkCudaErrorNoSync()
 
#endif // HOST_DEBUG
 
#ifdef __CUDA_ARCH__
// hide from device code
#undef errorQuda
#define errorQuda(...)
 
#endif
 
#endif // _UTIL_QUDA_H