[FS-Offloading] Batch Lookup in C (#46713)

Signed-off-by: <>
Co-authored-by: Varun Sundar Rabindranath <varun-sundar-rabindranath@h100-01.nemg-001.lab.rdu2.dc.redhat.com>
This commit is contained in:
Varun Sundar Rabindranath
2026-06-29 09:28:32 -07:00
committed by GitHub
co-authored by Varun Sundar Rabindranath
parent 379acd4e4f
commit c8fb2963bd
5 changed files with 162 additions and 2 deletions
+2
View File
@@ -777,6 +777,7 @@ class precompiled_wheel_utils:
"vllm/vllm_flash_attn/_vllm_fa3_C.abi3.so",
"vllm/cumem_allocator.abi3.so",
"vllm/spinloop.abi3.so",
"vllm/fs_io_C.abi3.so",
# ROCm-specific libraries
"vllm/_rocm_C.abi3.so",
}
@@ -1104,6 +1105,7 @@ if _is_cuda() or _is_hip():
if sys.version_info >= (3, 11):
ext_modules.append(CMakeExtension(name="vllm.spinloop"))
ext_modules.append(CMakeExtension(name="vllm.fs_io_C"))
if _is_hip():
ext_modules.append(CMakeExtension(name="vllm._rocm_C"))