Remove configure in bwd kernel launch

ea8a25ca · Tri Dao · af01244d · ea8a25ca
Commit ea8a25ca authored Jan 21, 2024 by Tri Dao
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

csrc/flash_attn/src/static_switch.h csrc/flash_attn/src/static_switch.h +1 -1

No files found.
--- a/csrc/flash_attn/src/static_switch.h
+++ b/csrc/flash_attn/src/static_switch.h
@@ -36,7 +36,7 @@
    }                                        \
  }()

-#define FWD_HEADDIM_SWITCH(HEADDIM, ...)   \
+#define HEADDIM_SWITCH(HEADDIM, ...)   \
  [&] {                                    \
    if (HEADDIM <= 32) {                   \
      constexpr static int kHeadDim = 32;  \