mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-13 16:37:04 +08:00
5a8ad57163
* Rename FusedOps to TernaryOps * Support ternary broadcast * Add where llop and mlop * Make where op work in cstyle codegen * Don't skip test_inf_where * Add backward path to where op * Use bool in cstyle codegen * Add LLVM where op * Add numpy where op * Add torch where op * Simplify where mlop * Update documentation * Forgot a rename * Merged relevant changes from PR #1195 onto PR #1196 * Add test to cover changes to linearizer.ast_parse for WHERE op Without this METAL will try to use ternary op on float4 and fail * Make where op work in wgsl backend * Allow ternary ops to be merged * Make mypy happy --------- Co-authored-by: Francis Lam <flam@alum.mit.edu>
Welcome to the tinygrad documentation!
Here you will find documentation for tinygrad, as well as some examples and tutorials.
Getting Started
Read the quick start guide here.
Or if you want to jump right in to how tinygrad works, you can read the abstraction stack documentation.
Or if you want to see some examples, you can look at the examples in the examples directory.
Or if you just want to see some of the things tinygrad can do, check out the showcase.
API
This is currently a big work in progress.
Resources
Environment Variables
Adding New Accelerators
Community
Contributing
The documentation mainly follows the core contributing guidelines in the README.md.
Additionally, we always welcome documentation contributions, especially for features that are currently under documented.