Skip to content

Commit 84aaa3d

Browse files
authored
ci: add apache-tvm-ffi to ci docker container (#1763)
<!-- .github/pull_request_template.md --> ## πŸ“Œ Description Fix the CI of #1641 ## πŸ” Related Issues #1641 ## πŸš€ 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. - [ ] All tests are passing (`unittest`, etc.). ## Reviewer Notes <!-- Optional: anything you'd like reviewers to focus on, concerns, etc. -->
1 parent f2b455b commit 84aaa3d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

β€Ždocker/install/install_python_packages.shβ€Ž

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ CUDA_VERSION=${1:-cu128}
2424

2525
pip3 install torch --index-url https://download.pytorch.org/whl/${CUDA_VERSION}
2626
pip3 install requests ninja pytest numpy scipy build nvidia-ml-py cuda-python einops nvidia-nvshmem-cu12
27+
pip3 install 'apache-tvm-ffi>=0.1.0b6'
2728
pip3 install nvidia-cutlass-dsl
2829
pip3 install 'nvidia-cudnn-frontend>=1.13.0'
2930

0 commit comments

Comments
Β (0)