[batch] add macro indirection

Co-authored-by: Tobias Ribizel <[email protected]>
ginkgo-project · Oct 21, 2024 · 8258bc9 · 8258bc9
1 parent 981ae40
commit 8258bc9
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/core/solver/batch_dispatch.hpp b/core/solver/batch_dispatch.hpp
@@ -164,12 +164,16 @@ enum class log_type { simple_convergence_completion };
 }  // namespace log
 
 
-#define GKO_BATCH_INSTANTIATE_STOP(macro, ...)                          \
-    macro(__VA_ARGS__,                                                  \
-          ::gko::batch::solver::device::batch_stop::SimpleAbsResidual); \
-    template macro(                                                     \
-        __VA_ARGS__,                                                    \
-        ::gko::batch::solver::device::batch_stop::SimpleRelResidual)
+#define GKO_INDIRECT(...) __VA_ARGS__
+
+
+#define GKO_BATCH_INSTANTIATE_STOP(macro, ...)                               \
+    GKO_INDIRECT(                                                            \
+        macro(__VA_ARGS__,                                                   \
+              ::gko::batch::solver::device::batch_stop::SimpleAbsResidual)); \
+    template GKO_INDIRECT(                                                   \
+        macro(__VA_ARGS__,                                                   \
+              ::gko::batch::solver::device::batch_stop::SimpleRelResidual))
 
 #define GKO_BATCH_INSTANTIATE_PRECONDITIONER(macro, ...)                   \
     GKO_BATCH_INSTANTIATE_STOP(                                            \