Fix the compilation error of XFORMERS_CHECK when PyTorch is not insta…

…lled (#913) * fix log * fix --------- Co-authored-by: lijiajieli <[email protected]>
facebookresearch · Nov 2, 2023 · 6b10d18 · 6b10d18
1 parent 96d381c
commit 6b10d18
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/xformers/csrc/attention/cuda/fmha/gemm_kernel_utils.h b/xformers/csrc/attention/cuda/fmha/gemm_kernel_utils.h
@@ -91,10 +91,17 @@
     std::cerr << #PTR " is not correctly aligned\n"; \
     return false;                                    \
   }
-#define XFORMERS_CHECK(COND, ERR)                       \
-  if (!(COND)) {                                        \
-    std::cerr << "'" #COND "' failed: " << ERR << "\n"; \
-    return false;                                       \
+
+template <typename... Args>
+void print_check_args(const Args&... args) {
+  (std::cerr << ... << args) << std::endl;
+}
+
+#define XFORMERS_CHECK(COND, ...)        \
+  if (!(COND)) {                         \
+    std::cerr << "'" #COND "' failed: "; \
+    print_check_args(__VA_ARGS__);       \
+    return false;                        \
   }
 #endif