## 📌 Description
Since we added nvidia-nvshmem-cu12 in setup.py, we need to install it in
the CI container image so unit tests can run when the module is loaded.
## 🔍 Related Issues
Related #1260
## 🚀 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.).
## Reviewer Notes
<!-- Optional: anything you'd like reviewers to focus on, concerns, etc.
-->
Signed-off-by: Emilien Macchi <emacchi@redhat.com>
<!-- .github/pull_request_template.md -->
## 📌 Description
cuda-python is widely used for communication kernels and so on and
should be part of our CI docker image.
## 🚀 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.).
Co-authored-by: yzh119 <zihaoy@nvidia.com>
<!-- .github/pull_request_template.md -->
## 📌 Description
Install the python packages for CI docker: mpi4py, pynvml. They will be
used for the comm ops.
## 🔍 Related Issues
https://github.com/flashinfer-ai/flashinfer/pull/1145,
https://github.com/flashinfer-ai/flashinfer/pull/1134
## 🚀 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
- [ ] 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.
-->
Co-authored-by: yongwww <yowu@nvidia.com>
The initial CI only includes a subset of unittests.
We will further reduce the test time and add all tests in later PRs.
