"docs/en/get_started/api_reference.md" did not exist on "2754cb11d5313e4ae375e64a791794b64bc29a71"
Commit 4d6d778f authored by sxtyzhangzk's avatar sxtyzhangzk
Browse files

[major] fix M_LOG2E missing on Windows

parent a8914127
......@@ -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")
......
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