chenyuandGitHub 8c6299bced move hand_coded_optimizations to heuristic.py [pr] (#9844) ...

* move hand_coded_optimizations to heuristic.py [pr]

also folded all long lines

* make a copy and rename self -> k

* fix test

2025-04-10 23:40:16 -04:00

..

codegen

move hand_coded_optimizations to heuristic.py [pr] (#9844)

2025-04-10 23:40:16 -04:00

engine

move hand_coded_optimizations to heuristic.py [pr] (#9844)

2025-04-10 23:40:16 -04: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

give _apply_map_to_tensors substitutes name [pr] (#9840)

2025-04-11 10:38:57 +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