Skip to content

GitLab

"llm/vscode:/vscode.git/clone" did not exist on "939c60473f6f8783e31a055c2847caa6099f3e2c"
Commit 2211db5f authored by Eric Engelhart's avatar Eric Engelhart
Browse files

Fixed switch statement, thanks @yocabon

parent 9b1b011b
Show 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 @@
#define FP16_SWITCH(COND, ...) \
[&] { \
if (COND) { \
using elem_type = std::conditional<true, __nv_bfloat16, __half>::type; \
using elem_type = __nv_bfloat16; \
return __VA_ARGS__(); \
} else { \
using elem_type = std::conditional<true, __nv_bfloat16, __half>::type; \
using elem_type = __half; \
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