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(), 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


#endif // _UTIL_QUDA_H