Skip to content

Commit

Permalink
Add comments to the newly added MMA macros.
Browse files Browse the repository at this point in the history
  • Loading branch information
hummingtree committed Dec 4, 2024
1 parent 979ba1a commit be3d180
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions include/quda_define.h.in
Original file line number Diff line number Diff line change
Expand Up @@ -138,27 +138,47 @@
#define GPU_MULTIGRID
#endif

/**
* @def QUDA_MULTIGRID
* @brief This macro is set when we have MMA enabled for the CUDA targets
*/
#cmakedefine QUDA_ENABLE_MMA

#ifdef QUDA_MULTIGRID

#cmakedefine QUDA_MULTIGRID_MMA_SETUP_TYPE
#ifdef QUDA_MULTIGRID_MMA_SETUP_TYPE
/**
* @def QUDA_MULTIGRID_SETUP_*
* @brief This macro is used to set the MMA type used for multigrid setup
*/
#define QUDA_MULTIGRID_MMA_SETUP_@QUDA_MULTIGRID_MMA_SETUP_TYPE@
#endif

#cmakedefine QUDA_MULTIGRID_MMA_DSLASH_TYPE
#ifdef QUDA_MULTIGRID_MMA_DSLASH_TYPE
/**
* @def QUDA_MULTIGRID_MMA_DSLASH_*
* @brief This macro is used to set the MMA type used for coarse dslash
*/
#define QUDA_MULTIGRID_MMA_DSLASH_@QUDA_MULTIGRID_MMA_DSLASH_TYPE@
#endif

#cmakedefine QUDA_MULTIGRID_MMA_PROLONGATOR_TYPE
#ifdef QUDA_MULTIGRID_MMA_PROLONGATOR_TYPE
/**
* @def QUDA_MULTIGRID_MMA_PROLONGATOR_*
* @brief This macro is used to set the MMA type used for prolongator
*/
#define QUDA_MULTIGRID_MMA_PROLONGATOR_@QUDA_MULTIGRID_MMA_PROLONGATOR_TYPE@
#endif

#cmakedefine QUDA_MULTIGRID_MMA_RESTRICTOR_TYPE
#ifdef QUDA_MULTIGRID_MMA_RESTRICTOR_TYPE
/**
* @def QUDA_MULTIGRID_MMA_RESTRICTOR_*
* @brief This macro is used to set the MMA type used for restrictor
*/
#define QUDA_MULTIGRID_MMA_RESTRICTOR_@QUDA_MULTIGRID_MMA_RESTRICTOR_TYPE@
#endif

Expand Down

0 comments on commit be3d180

Please sign in to comment.