We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a5450f1 commit d606cfdCopy full SHA for d606cfd
csrc/cpu/cpu_types_x86.hpp
@@ -19,6 +19,7 @@ namespace vec_op {
19
#define VLLM_DISPATCH_CASE_FLOATING_TYPES_FP8(...) \
20
AT_DISPATCH_CASE(at::ScalarType::Float, __VA_ARGS__) \
21
AT_DISPATCH_CASE(at::ScalarType::BFloat16, __VA_ARGS__) \
22
+ AT_DISPATCH_CASE(at::ScalarType::Half, __VA_ARGS__) \
23
AT_DISPATCH_CASE(at::ScalarType::Float8_e5m2, __VA_ARGS__)
24
25
#define VLLM_DISPATCH_FLOATING_TYPES(TYPE, NAME, ...) \
0 commit comments