| .. |
|
codegen
|
move hand_coded_optimizations to heuristic.py [pr] (#9844)
|
2025-04-10 23:40:16 -04:00 |
|
engine
|
unbind variables when creating ScheduleItems [pr] (#9846)
|
2025-04-11 15:23:53 +08:00 |
|
frontend
|
add onnx frontend stub [pr] (#9558)
|
2025-03-24 12:24:34 +08:00 |
|
nn
|
nn.state docs cleanup (#8332)
|
2025-03-18 17:16:40 -04:00 |
|
renderer
|
label ranges with their number (#9805)
|
2025-04-09 14:31:18 +08:00 |
|
runtime
|
fix am driver for gfx1201 (#9836)
|
2025-04-10 19:33:02 +03:00 |
|
shape
|
cache folded upcast [pr] (#9733)
|
2025-04-04 11:23:19 +08:00 |
|
viz
|
add err handling tests to viz + cleanups (#9825)
|
2025-04-10 14:05:05 +08:00 |
|
__init__.py
|
UOp.variable (#7010)
|
2024-10-12 18:20:44 +08:00 |
|
device.py
|
canonicalize Device.DEFAULT (#9835)
|
2025-04-10 23:02:11 +08:00 |
|
dtype.py
|
Revert "FP8 support on NVIDIA (#8631)"
|
2025-04-09 12:27:41 +08:00 |
|
gradient.py
|
Ops.FUSE makes softmax a single kernel (#9808)
|
2025-04-09 22:56:28 +08:00 |
|
helpers.py
|
use functools.cache instead of lru_cache(None) [pr] (#9714)
|
2025-04-03 11:47:13 +08:00 |
|
ops.py
|
fix single kernel softmax with cast (#9842)
|
2025-04-11 12:12:02 +08:00 |
|
py.typed
|
add a single py.typed (#6083)
|
2024-08-14 17:31:46 -07:00 |
|
spec.py
|
fix infinite loop in flash attention (#9827)
|
2025-04-10 20:06:44 +08:00 |
|
tensor.py
|
give _apply_map_to_tensors substitutes name [pr] (#9840)
|
2025-04-11 10:38:57 +08:00 |
