upd

Author: yzh119

pyproject.toml

@@ -43,10 +43,12 @@ py-modules = ["build_backend", "build_utils"]
 
 [tool.setuptools.dynamic]
 version = {attr = "flashinfer._build_meta.__version__"}
+dependencies = {file = ["requirements.txt"]}
 
 [tool.setuptools.packages.find]
 where = ["."]
 include = ["flashinfer*"]
+exclude = ["flashinfer-jit-cache*", "flashinfer-cubin*"]
 
 [tool.setuptools.package-dir]
 "flashinfer.data" = "."

requirements.txt

@@ -0,0 +1,13 @@
+apache-tvm-ffi==0.1.0b15
+click
+einops
+ninja
+numpy
+nvidia-cudnn-frontend>=1.13.0
+nvidia-cutlass-dsl>=4.2.1
+nvidia-ml-py
+packaging>=24.2
+requests
+tabulate
+torch
+tqdm