File tree 1 file changed +13
-13
lines changed
1 file changed +13
-13
lines changed Original file line number Diff line number Diff line change @@ -51,20 +51,20 @@ using namespace flashinfer;
51
51
} \
52
52
}()
53
53
#else
54
- #define DISPATCH_PYTORCH_DTYPE_TO_CTYPE (pytorch_dtype, c_type, ...) [&]() -> bool { \
55
- switch (pytorch_dtype) { \
56
- case at::ScalarType::Half: { \
57
- using c_type = nv_half; \
58
- return __VA_ARGS__ (); \
59
- } \
60
- default : \
61
- std::ostringstream oss; \
54
+ #define DISPATCH_PYTORCH_DTYPE_TO_CTYPE (pytorch_dtype, c_type, ...) \
55
+ [&]() -> bool { \
56
+ switch (pytorch_dtype) { \
57
+ case at::ScalarType::Half: { \
58
+ using c_type = nv_half; \
59
+ return __VA_ARGS__ (); \
60
+ } \
61
+ default : \
62
+ std::ostringstream oss; \
62
63
oss << __PRETTY_FUNCTION__ << " failed to dispatch data type " << pytorch_dtype; \
63
- TORCH_CHECK (false , oss.str ()); \
64
- return false ; \
65
- }
66
- }
67
- ()
64
+ TORCH_CHECK (false , oss.str ()); \
65
+ return false ; \
66
+ } \
67
+ } ()
68
68
#endif
69
69
70
70
#ifdef FLASHINFER_ENABLE_FP8
You can’t perform that action at this time.
0 commit comments