Skip to content

Commit d66181a

Browse files
feat(lora) Add lora bgmv_shrink & bgmv_expand kernels (#31)
* feat(lora): Add lora bgmv_shrink && bgmv_expand ops Signed-off-by: chzhang <[email protected]> * feat(lora): Add lora bgmv_shrink && bgmv_expand ops Signed-off-by: chzhang <[email protected]> * feat(lora): Add lora bgmv_shrink && bgmv_expand ops Signed-off-by: chzhang <[email protected]> * feat(lora): Add lora bgmv_shrink && bgmv_expand ops Signed-off-by: chzhang <[email protected]> --------- Signed-off-by: chzhang <[email protected]>
1 parent 2dcde10 commit d66181a

File tree

14 files changed

+3064
-2
lines changed

14 files changed

+3064
-2
lines changed

CMakeLists.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -183,6 +183,8 @@ define_gpu_extension_target(
183183
if(VLLM_GPU_LANG STREQUAL "SYCL")
184184
set(VLLM_EXT_XPU_SRC
185185
"csrc/xpu/torch_bindings.cpp"
186+
"csrc/xpu/lora/lora_shrink.cpp"
187+
"csrc/xpu/lora/lora_expand.cpp"
186188
)
187189
include_directories("/usr/include")
188190
set(CMPLR_ROOT $ENV{CMPLR_ROOT})

0 commit comments

Comments
 (0)