George HotzandGitHub 599eb266b1 optionally use a copy kernel instead of SDMA (#4116) ...

* optionally use a copy kernel

* lazyops in copied kernels

* add sync

* no sdma at all

* work

* copy_ast

2024-04-12 23:10:41 -07:00

..

codegen

optionally use a copy kernel instead of SDMA (#4116)

2024-04-12 23:10:41 -07:00

engine

optionally use a copy kernel instead of SDMA (#4116)

2024-04-12 23:10:41 -07:00

features

move sum acc_dtype into lazy so it applies to backward (#4149)

2024-04-11 14:43:56 -04:00

nn

Use exec_alu for lazy const folding (#4039)

2024-04-02 20:52:05 -04:00

renderer

Update ssa input order and annotate types in cstyle and assembly (#4117)

2024-04-09 13:10:29 -04:00

runtime

hotfix: CUDA_P2P works (#4155)

2024-04-12 18:20:12 +03:00

shape

assert if expr_idxs return might be outside of int32 (#4157)

2024-04-12 14:18:35 -04:00

__init__.py

spend 5 lines to bring mnist into the repo (#4122)

2024-04-09 19:24:57 -07:00

buffer.py

use Buffer.ensure_allocated in search _ensure_buffer_alloc (#4132)

2024-04-10 13:11:50 -04:00

device.py

rewrite the jit in the context of new schedule (#4162)

2024-04-12 21:54:36 -07:00

dtype.py

rename Scalar to ConstType and cast_scalar to as_const (#3946)

2024-03-26 22:39:58 -04:00

function.py

move sum acc_dtype into lazy so it applies to backward (#4149)

2024-04-11 14:43:56 -04:00

helpers.py

PADTO SUM if parents of sum are all zero-preserving (#4140)

2024-04-10 22:16:12 -04:00

lazy.py

optionally use a copy kernel instead of SDMA (#4116)

2024-04-12 23:10:41 -07:00

ops.py

optionally use a copy kernel instead of SDMA (#4116)

2024-04-12 23:10:41 -07:00

tensor.py

cleanup lbs (#4163)

2024-04-12 22:32:16 -07:00