Fixed non-windows nvshmem import and wrong spacy import

Signed-off-by: Javier <25750030+SystemPanic@users.noreply.github.com>

Author: SystemPanic

flashinfer/jit/cpp_ext.py

@@ -9,7 +9,6 @@
 from typing import List, Optional
 
 import torch
-from spacy.compat import is_windows
 from torch.utils.cpp_extension import (
     _TORCH_PATH,
     CUDA_HOME,
@@ -65,6 +64,7 @@ def generate_ninja_build_for_op(
             common_cflags.append(f"-I{dir.resolve()}")
 
     is_windows = platform.system() == "Windows"
+
     if is_windows:
         for dir in system_includes:
             common_cflags.append(f"-I{dir}")
@@ -89,7 +89,6 @@ def generate_ninja_build_for_op(
 
 
     common_cuda_flags = common_cflags.copy()
-    is_windows = platform.system() == "Windows"
 
     if is_windows:
         common_cuda_flags = [

setup.py

@@ -86,8 +86,11 @@ def generate_build_meta(aot_build_meta: dict) -> None:
     "requests",
     "cuda-python",
     "pynvml",
-    "einops",
+    "einops"
 ]
+if not IS_WINDOWS:
+    install_requires.append("nvidia-nvshmem-cu12")
+
 generate_build_meta({})
 
 if IS_WINDOWS:
@@ -163,4 +166,4 @@ def get_cuda_version() -> Version:
     cmdclass=cmdclass,
     install_requires=install_requires,
     options={"bdist_wheel": {"py_limited_api": "cp39"}}
-)
+)