nimlgenandGitHub 8ef5490ec8 cuda tranfer + async copyin (#3873)

2024-03-22 09:01:37 -07:00

..

codegen

PTX - implement float 4, ptr arithmetics and other speed improvements (#3775)

2024-03-22 08:54:02 -07:00

features

proposal: multioutput JIT spec (#3856)

2024-03-21 21:28:30 -07:00

nn

hotfix: this makes beautiful mnist work again, not okay

2024-03-18 18:22:44 -07:00

renderer

PTX - implement float 4, ptr arithmetics and other speed improvements (#3775)

2024-03-22 08:54:02 -07:00

runtime

cuda tranfer + async copyin (#3873)

2024-03-22 09:01:37 -07:00

shape

fix View.pad arg type (#3845)

2024-03-20 19:36:02 -04:00

__init__.py

move graph.py and jit.py into features (#3376)

2024-02-12 17:34:34 +01:00

device.py

log optimized kernels and a script to compare with non-optimized ones (#3829)

2024-03-20 19:22:08 -04:00

dtype.py

Rename .sz to .count on DType (#3413)

2024-02-15 15:03:49 +01:00

helpers.py

Ring allreduce try 2 (#3852)

2024-03-21 19:17:51 -04:00

lazy.py

Ring allreduce try 2 (#3852)

2024-03-21 19:17:51 -04:00

mlops.py

reciprocal mlop (#3694)

2024-03-11 16:08:46 -07:00

ops.py

simple LoadOps.ASSIGN (#3745)

2024-03-14 20:44:34 -07:00

realize.py

cuda tranfer + async copyin (#3873)

2024-03-22 09:01:37 -07:00

tensor.py

user _resolve_dim in argmax (#3846)

2024-03-20 20:17:30 -04:00