chenyu ce454793e6 support specifying dtype for Tensor.linear (#9886)

2025-04-14 13:55:11 -04:00

..

codegen

clean up some patterns [pr] (#9880)

2025-04-14 11:33:22 +01:00

engine

add names to grouper rewrites + cleanups [pr] (#9881)

2025-04-14 19:47:36 +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

add arch to AMDLLVMRenderer (#9884)

2025-04-14 19:59:22 +03:00

runtime

add arch to AMDLLVMRenderer (#9884)

2025-04-14 19:59:22 +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

fast pattern matcher [pr] (#9737)

2025-04-14 15:24:41 +01: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

support specifying dtype for Tensor.linear (#9886)

2025-04-14 13:55:11 -04:00

upat.py

fast pattern matcher [pr] (#9737)

2025-04-14 15:24:41 +01:00