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
norm
vllm
Commits
614856da
"vscode:/vscode.git/clone" did not exist on "9cca3e0b037a5b86a839caf4dcdc6a177fbb892c"
Unverified
Commit
614856da
authored
Dec 14, 2023
by
Mingcan Xiang
Committed by
GitHub
Dec 14, 2023
Browse files
Avoid multiple redefinition (#1817)
parent
05bdf4ea
Changes
4
Show whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
8 additions
and
0 deletions
+8
-0
csrc/cache.h
csrc/cache.h
+2
-0
csrc/cuda_utils.h
csrc/cuda_utils.h
+2
-0
csrc/dispatch_utils.h
csrc/dispatch_utils.h
+2
-0
csrc/ops.h
csrc/ops.h
+2
-0
No files found.
csrc/cache.h
View file @
614856da
#pragma once
#include <torch/extension.h>
#include <torch/extension.h>
#include <map>
#include <map>
...
...
csrc/cuda_utils.h
View file @
614856da
#pragma once
#include <torch/extension.h>
#include <torch/extension.h>
int
get_device_attribute
(
int
get_device_attribute
(
...
...
csrc/dispatch_utils.h
View file @
614856da
...
@@ -2,6 +2,8 @@
...
@@ -2,6 +2,8 @@
* Adapted from
* Adapted from
* https://github.com/pytorch/pytorch/blob/v2.0.1/aten/src/ATen/Dispatch.h
* https://github.com/pytorch/pytorch/blob/v2.0.1/aten/src/ATen/Dispatch.h
*/
*/
#pragma once
#include <torch/extension.h>
#include <torch/extension.h>
#define VLLM_DISPATCH_CASE_FLOATING_TYPES(...) \
#define VLLM_DISPATCH_CASE_FLOATING_TYPES(...) \
...
...
csrc/ops.h
View file @
614856da
#pragma once
#include <torch/extension.h>
#include <torch/extension.h>
void
paged_attention_v1
(
void
paged_attention_v1
(
...
...
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