20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Avoid multiple redefinition (#1817)

Browse Source
This commit is contained in:
Mingcan Xiang 2023-12-14 12:35:58 -05:00 committed by GitHub
parent 05bdf4eaf3
commit 614856da25
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
csrc/cache.h
View File

@ -1,3 +1,5 @@
#pragma once
#include <torch/extension.h> #include <torch/extension.h>
#include <map> #include <map>

2
csrc/cuda_utils.h
View File

@ -1,3 +1,5 @@
#pragma once
#include <torch/extension.h> #include <torch/extension.h>
int get_device_attribute( int get_device_attribute(

2
csrc/dispatch_utils.h
View File

@ -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(...) \

2
csrc/ops.h
View File

@ -1,3 +1,5 @@
#pragma once
#include <torch/extension.h> #include <torch/extension.h>
void paged_attention_v1( void paged_attention_v1(