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
- [FIX][Relax] Skip ExpandMatmulOfSum when addition broadcasts
- [Fix][Relax] Preserve zero extents in ONNX import and reshape roundtrips
- [Bug] ExpandMatmulOfSum silently miscompiles matmul(x, a+b) when addends broadcast
- [Fix][Runtime] Keep sliding-window page metadata separate from global attention
- [FIX][Relax] Infer call_te output vdevice from keyword arguments
- [Relax][Frontend][TFLite] Add DYNAMIC_UPDATE_SLICE operator mapping
- [Bug] CUDA workspace cleanup aborts on Windows at thread/process exit
- [Bug][Relax][CUDA/NVPTX] `R.median` legalizes to the CPU `tvm.contrib.sort.argsort` extern, which ends up inside a device kernel: `Unresolved call tirx.call_packed` (cuda) / `unknown intrinsic tirx.call_packed` (nvptx)
- [Bug][Vulkan] SPIR-V codegen ICHECKs on bool buffers: storing into a bool kernel parameter ("bool vs. int8") and using a local bool buffer ("int8 vs. bool")
- [Bug][Relax][CUDA/NVPTX] `R.vision.all_class_non_max_suppression` fails GPU compilation in `FlattenTensor`: "Tensor T_gather is used before its definition"
- Docs
- Python not yet supported