Skip to content

Commit 60ae234

Browse files
authored
fix: add missing 'requests' when building the package with AOT (#1517)
## πŸ“Œ Description When building flashinfer-python with AOT enabled, we need the 'requests' package otherwise it fails with missing packages when importing modules. ## πŸ” Related Issues Fixes #1516 ## πŸš€ Pull Request Checklist Thank you for contributing to FlashInfer! Before we review your pull request, please make sure the following items are complete. ### βœ… Pre-commit Checks - [x] I have installed `pre-commit` by running `pip install pre-commit` (or used your preferred method). - [x] I have installed the hooks with `pre-commit install`. - [x] I have run the hooks manually with `pre-commit run --all-files` and fixed any reported issues. > If you are unsure about how to set up `pre-commit`, see [the pre-commit documentation](https://pre-commit.com/). ## πŸ§ͺ Tests - [x] Tests have been added or updated as needed. - [x] All tests are passing (`unittest`, etc.). Signed-off-by: Emilien Macchi <[email protected]>
1 parent 96e2140 commit 60ae234

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

β€Žcustom_backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
_aot_ops_dir = _root / "aot-ops"
99
_aot_ops_package_dir = _root / "build" / "aot-ops-package-dir"
1010

11-
_requires_for_aot = ["torch", "ninja", "numpy"]
11+
_requires_for_aot = ["torch", "ninja", "numpy", "requests"]
1212

1313

1414
def _rm_aot_ops_package_dir():

0 commit comments

Comments
Β (0)