4 people d97d5a7689 Optimize PTX gated loads index calculation (#4304) ...

* WIP but working

* Cleanup

* Remove float4 pred and alt

* Cleanup

* this is somehow slowin it down

* Simplify

* add define var to ignore when optimizing gates

* Update assembly.py

* Test for optimizing gated loads

* Cleanup

* Fix NEG needed before if

* Remove unused parameters

* Update assembly.py

* Fix for cachable gone

---------

Co-authored-by: oz <[email protected]>
Co-authored-by: chenyu <[email protected]>

2024-05-13 10:14:01 -07:00

..

codegen

all uops are now cachable (#4564)

2024-05-12 22:34:35 -07:00

engine

small scheduler refactor (#4569)

2024-05-13 10:47:39 +03:00

features

move scheduleitem to schedule.py (#4541)

2024-05-11 21:13:04 -07:00

nn

quantized llama multilazybuffer fix (#4557)

2024-05-12 14:19:21 -07:00

renderer

Optimize PTX gated loads index calculation (#4304)

2024-05-13 10:14:01 -07:00

runtime

simple stuff from new_uops branch (#4563)

2024-05-12 22:18:05 -07:00

shape

add consecutive property to shapetracker

2024-05-02 10:41:28 -07:00

__init__.py

add mnist tutorial

2024-04-25 16:08:32 +08:00

device.py

all realize 2 (#4527)

2024-05-10 22:43:09 -07:00

dtype.py

move acc_dt out of lazy (#4382)

2024-05-06 07:41:25 -07:00

function.py

move acc_dt out of lazy (#4382)

2024-05-06 07:41:25 -07:00

helpers.py

add cpu objdump to LLVM/CLANG (#4537)

2024-05-11 14:28:44 -07:00

lazy.py

bring buffer back to device (#4517)

2024-05-10 11:22:31 -07:00

ops.py

move scheduleitem to schedule.py (#4541)

2024-05-11 21:13:04 -07:00

tensor.py

cleanup dtype of tensor creation from list (#4566)

2024-05-13 02:47:41 -04:00