[major] fix M_LOG2E missing on Windows

a8914127 · sxtyzhangzk · 1fe94628 · a8914127
Commit a8914127 authored Nov 10, 2024 by sxtyzhangzk
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

csrc/block_sparse_attn/src/softmax.h csrc/block_sparse_attn/src/softmax.h +4 -0

No files found.
--- a/csrc/block_sparse_attn/src/softmax.h
+++ b/csrc/block_sparse_attn/src/softmax.h
@@ -16,6 +16,10 @@
 #include "philox.cuh"
 #include "utils.h"

+#ifndef M_LOG2E
+#define M_LOG2E 1.4426950408889634074
+#endif
+
 namespace flash {

 using namespace cute;