Barath c5aea13a65 Fix evaluation stage in examples/transformer.py when using CUDA (#1150) ...

* make test data as contiguous array

* standardise contiguous array for all input data in cuda ops

* swap to x.ravel

2023-07-06 18:07:10 -07:00

..

codegen

Remove POW llop and add SQRT llop (#1104)

2023-07-05 18:07:58 -07:00

nn

[perf] Replace more list comprehension with * (#1106)

2023-07-03 10:49:23 -07:00

runtime

Fix evaluation stage in examples/transformer.py when using CUDA (#1150)

2023-07-06 18:07:10 -07:00

shape

from tensor cores + lb touchup (#1127)

2023-07-04 15:45:20 -07:00

graph.py

Optimizations in lazy.py (#987)

2023-06-26 13:55:42 -07:00

helpers.py

Remove float64 (#1101)

2023-07-04 08:40:51 -07:00

jit.py

HIP backend (#750)

2023-06-18 11:35:57 -07:00

lazy.py

Fold expand preceding reduce if the reduction is on the same axis as the expansion (#1134)

2023-07-06 13:41:05 -07:00

mlops.py

Remove POW llop and add SQRT llop (#1104)

2023-07-05 18:07:58 -07:00

ops.py

Remove POW llop and add SQRT llop (#1104)

2023-07-05 18:07:58 -07:00

state.py

Whisper (#919)

2023-06-03 18:55:14 -07:00

tensor.py

Don't save parents unless needed (#1142)

2023-07-05 18:11:57 -07:00