tvm
https://github.com/apache/tvm
Python
Open deep learning compiler stack for cpu, gpu and specialized accelerators
Triage Issues!
When you volunteer to triage issues, you'll receive an email each day with a link to an open issue that needs help in this project. You'll also receive instructions on how to triage issues.
Triage Docs!
Receive a documented method or class from your favorite GitHub repos in your inbox every day. If you're really pro, receive undocumented methods or classes and supercharge your commit history.
Python not yet supported3 Subscribers
Add a CodeTriage badge to tvm
Help out
- Issues
- [Bug] PRelu with 1-D input fails to import in Relax: relax.nn.prelu uses axis=1 out of range
- [Bug] PRelu import/compile fails when slope initializer is broadcastable rank-2 (1,1): topi.nn.prelu requires 1-D slope
- [Bug] Segfault in `tvm.compile` on **LLVM (CPU) target** when `tir.ptx_ldg32=1`: unexpectedly runs `tir::transform::InjectPTXLDG32` / `PTXRewriter` and crashes in `BufferStore`
- Bump urllib3 from 1.26.18 to 2.6.3 in /docker/python
- [Bug] Segfault in LLVM COFFOptTable when combining TVM CUDA target with PyTorch Lightning + torchmetrics in the same process
- [Bug] Segmentation fault when using TVM together with `transformers` (flan-t5-base + bfloat16 + `use_cache=True`)
- [Bug] Segmentation fault when importing `torchmetrics` after creating a TVM CUDA target (LLVM initialization / COFF OptTable)
- [RELAX][LAYOUT] Support multiple axis packing
- Bump virtualenv from 20.14.1 to 20.36.1 in /docker/python
- Bump filelock from 3.7.0 to 3.20.3 in /docker/python
- Docs
- Python not yet supported