| .. |
|
codegen
|
use UOp.st for kernel reduce axes (#8499)
|
2025-01-13 06:24:11 -05:00 |
|
engine
|
remove cast before view (#8613)
|
2025-01-14 15:04:58 -05:00 |
|
nn
|
minor cleanups in state.py [pr] (#8438)
|
2024-12-28 20:13:25 -05:00 |
|
renderer
|
CUDA Turing TC (#8597)
|
2025-01-14 10:35:14 -08:00 |
|
runtime
|
CUDA Turing TC (#8597)
|
2025-01-14 10:35:14 -08:00 |
|
shape
|
use unravel in views_to_indexed_uops [pr] (#8560)
|
2025-01-12 10:25:55 -05:00 |
|
viz
|
viz fixups + scheduler option [pr] (#8557)
|
2025-01-10 15:09:31 -08:00 |
|
__init__.py
|
UOp.variable (#7010)
|
2024-10-12 18:20:44 +08:00 |
|
device.py
|
free intermediate buffers in the jit [pr] (#8581)
|
2025-01-12 15:41:41 -08:00 |
|
dtype.py
|
typos found by gemini [pr] (#8400)
|
2024-12-24 22:32:25 -05:00 |
|
function.py
|
Tensor.mod (#8458)
|
2024-12-31 11:31:42 -05:00 |
|
gradient.py
|
compute gradient [pr] (#8237)
|
2024-12-13 20:46:01 -08:00 |
|
helpers.py
|
use CAPTURE_PROCESS_REPLAY=1 in CI [pr] (#8564)
|
2025-01-11 06:03:48 -05:00 |
|
multi.py
|
raise RuntimeError for uneven shard [pr] (#8593)
|
2025-01-14 14:51:48 -05:00 |
|
ops.py
|
remove cast before view (#8613)
|
2025-01-14 15:04:58 -05:00 |
|
py.typed
|
add a single py.typed (#6083)
|
2024-08-14 17:31:46 -07:00 |
|
tensor.py
|
raise RuntimeError for uneven shard [pr] (#8593)
|
2025-01-14 14:51:48 -05:00 |
bfbe81df71