Moving branch to a different repo

neuralmagic · Aug 2, 2024 · 5a2ab25 · 5a2ab25
1 parent c8a7e93
commit 5a2ab25
Show file tree

Hide file tree

Showing 12 changed files with 3,860 additions and 101 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -211,7 +211,8 @@ define_gpu_extension_target(
 
 set(VLLM_MOE_EXT_SRC
   "csrc/moe/torch_bindings.cpp"
-  "csrc/moe/topk_softmax_kernels.cu")
+  "csrc/moe/topk_softmax_kernels.cu"
+  "csrc/moe/marlin_moe_ops.cu")
 
 define_gpu_extension_target(
   _moe_C