add vxpu

2026-02-05 19:36:06 +08:00
parent dc63e81a7f
commit 4d8575115a
131 changed files with 28540 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -19,6 +19,24 @@ ext_modules = [
        ],
        library_dirs=["/usr/local/cuda/lib64"],
        extra_compile_args=['-O3'],
+    ),
+    CppExtension(
+        name='vllm_kunlun._kunlun_vxpu',
+        sources=[
+            'vllm_kunlun/csrc/xpumem_allocator.cpp',
+            'vllm_kunlun/csrc/vxpu_offload/shm_worker.cpp',
+        ],
+        include_dirs=[
+            os.path.join(ROOT_DIR, 'vllm_kunlun/csrc/vxpu_offload/include'),
+            '/opt/vllm_kunlun/xcudart/include',
+            '/usr/local/cuda/include',
+        ],
+        library_dirs=[
+            "/usr/local/cuda/lib64",
+            "/opt/vllm_kunlun/xcudart/lib",
+        ],
+        libraries=["xpurt", "rt"],
+        extra_compile_args=['-O3'],
    )
 ]