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
- [Relax][PyTorch] Add support for gather, flip and take ops
- [Refactor] Introduce base Executable class and `tvm.compile` interface
- [Runtime][Target] Add C module for host
- [WebGPU] Support warp-level shuffle primitives with subgroup
- [Bug] Cannot find -lxml2 when linking libtvm.so on Ubuntu 22.04 (WSL2)
- [BugFix][Codegen, CUDA] Fix faulty codegen for FP8
- BF16 Support
- [Bug] Error converting operator Cast, with inputs: [R.shape([x_0, x_1])]
- [Bug] InternalError: Check failed: (!expr->struct_info_.defined()) is false: To ensure idempotency, the expression passed to UpdateStructInfo must not have any prior StructInfo. However, expression # from tvm.script import tir as T @T.prim_func(private=True)
- [Bug] Deprecation warning displayed on tests
- Docs
- Python not yet supported