Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
zhangdong1
Block-Sparse-Attention
Commits
4d6d778f
"docs/en/get_started/api_reference.md" did not exist on "2754cb11d5313e4ae375e64a791794b64bc29a71"
Commit
4d6d778f
authored
Nov 10, 2024
by
sxtyzhangzk
Browse files
[major] fix M_LOG2E missing on Windows
parent
a8914127
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
0 deletions
+4
-0
csrc/block_sparse_attn/flash_api.cpp
csrc/block_sparse_attn/flash_api.cpp
+4
-0
No files found.
csrc/block_sparse_attn/flash_api.cpp
View file @
4d6d778f
...
...
@@ -28,6 +28,10 @@ using namespace pytorch_compat;
#include "src/flash.h"
#include "src/static_switch.h"
#ifndef M_LOG2E
#define M_LOG2E 1.4426950408889634074
#endif
#define CHECK_DEVICE(x) TORCH_CHECK(x.is_cuda(), #x " must be on CUDA")
#define CHECK_SHAPE(x, ...) TORCH_CHECK(x.sizes() == torch::IntArrayRef({__VA_ARGS__}), #x " must have shape (" #__VA_ARGS__ ")")
#define CHECK_CONTIGUOUS(x) TORCH_CHECK(x.is_contiguous(), #x " must be contiguous")
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment