Skip to content

Commit 4c4276a

Browse files
authored
Force upgrade cuDNN to latest (#1401)
<!-- .github/pull_request_template.md --> ## πŸ“Œ Description Force upgrade cuDNN to latest ## πŸ” Related Issues <!-- Link any related issues here --> ## πŸš€ 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 - [ ] I have installed `pre-commit` by running `pip install pre-commit` (or used your preferred method). - [ ] I have installed the hooks with `pre-commit install`. - [ ] 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 - [ ] 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 49d744d commit 4c4276a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

β€Žscripts/task_test_blackwell_kernels.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ set -x
66
: ${CUDA_VISIBLE_DEVICES:=0}
77

88
pip install -e . -v
9+
pip install --upgrade nvidia-cudnn-cu12
910

1011
# run task_blackwell_utils_kernels.sh
1112
bash scripts/run_test_blackwell_utils_kernels.sh

0 commit comments

Comments
Β (0)