We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents a70d31c + 2628beb commit 0e68a2fCopy full SHA for 0e68a2f
flashinfer-cubin/pyproject.toml
@@ -1,5 +1,5 @@
1
[build-system]
2
-requires = ["setuptools>=61.0", "wheel", "requests", "filelock", "torch", "tqdm", "numpy", "apache-tvm-ffi>=0.1,<0.2"]
+requires = ["setuptools>=61.0", "wheel", "requests", "filelock", "torch", "tqdm", "numpy", "apache-tvm-ffi>=0.1,<0.2", "nvidia-ml-py"]
3
build-backend = "build_backend"
4
backend-path = ["."]
5
0 commit comments