George HotzandGitHub bd5939514d clean up a few patterns [pr] (#9873)

2025-04-13 20:19:37 +01:00

..

codegen

clean up a few patterns [pr] (#9873)

2025-04-13 20:19:37 +01:00

engine

add asserts for KERNEL op ast [pr] (#9868)

2025-04-13 16:50:18 +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

Use $CUDA_PATH/include for CUDA headers (#9858)

2025-04-13 16:20:19 +01: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

clean up a few patterns [pr] (#9873)

2025-04-13 20:19:37 +01:00

tensor.py

give _apply_map_to_tensors substitutes name [pr] (#9840)

2025-04-11 10:38:57 +08:00