Skip to content

GitLab

Commit 2211db5f authored by Eric Engelhart's avatar Eric Engelhart
Browse files

Fixed switch statement, thanks @yocabon

parent 9b1b011b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
csrc/flash_attn/src/fp16_switch.h
View file @ 2211db5f
...@@ -18,10 +18,10 @@ ...@@ -18,10 +18,10 @@
#define FP16_SWITCH(COND, ...) \ #define FP16_SWITCH(COND, ...) \
[&] { \ [&] { \
if (COND) { \ if (COND) { \
using elem_type = std::conditional<true, __nv_bfloat16, __half>::type; \ using elem_type = __nv_bfloat16; \
return __VA_ARGS__(); \ return __VA_ARGS__(); \
} else { \ } else { \
using elem_type = std::conditional<true, __nv_bfloat16, __half>::type; \ using elem_type = __half; \
return __VA_ARGS__(); \ return __VA_ARGS__(); \
} \ } \
}() }()
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment