Update paths from include to libflashinfer/include

Author: diptorupd

custom_backend.py

@@ -37,6 +37,6 @@ def ln(src: str, dst: str) -> None:
 
     ln("3rdparty/cutlass", "cutlass")
     ln("csrc", "csrc")
-    ln("include", "include")
+    ln("libflashinfer/include", "include")
     ln("tvm_binding", "tvm_binding")
     return orig.build_editable(wheel_directory, config_settings, metadata_directory)

pyproject.toml

@@ -43,7 +43,7 @@ packages = [
     "flashinfer.data.csrc",
     "flashinfer.data.cutlass",
     "flashinfer.data.tvm_binding",
-    "flashinfer.data.include",
+    "flashinfer.data.libflashinfer.include",
     "flashinfer.jit",
     "flashinfer.jit.attention",
     "flashinfer.triton",
@@ -54,11 +54,12 @@ include-package-data = false
 [tool.setuptools.package-dir]
 "flashinfer.data" = "."
 "flashinfer.data.cutlass" = "3rdparty/cutlass"
+"flashinfer.data.libflashinfer" = "libflashinfer"
 
 [tool.setuptools.package-data]
 "flashinfer.data" = [
     "csrc/**",
-    "include/**",
+    "libflashinfer/include/**",
     "tvm_binding/**",
     "version.txt"
 ]

setup.py

@@ -209,7 +209,7 @@ def __init__(self, *args, **kwargs) -> None:
 
     cutlass = root / "3rdparty" / "cutlass"
     include_dirs = [
-        root.resolve() / "include",
+        root.resolve() / "libflashinfer/include",
         cutlass.resolve() / "include",  # for group gemm
         cutlass.resolve() / "tools" / "util" / "include",
     ]