Tobias Fischer 5326bbc9a6 fix causal mask in Tensor class (#2425) ...

* fixed causal mask in Tensor class

* added tests for scaled attention against pytorch

* cleaned up test formatting

* removed duplicate test

2023-11-24 18:38:18 -05:00

..

codegen

clean up ast_parse (#2428)

2023-11-24 16:43:32 -05:00

features

remove triton (#2396)

2023-11-23 07:40:59 -08:00

nn

torch and numpy don't share ops anymore (#2412)

2023-11-23 16:58:10 -08:00

renderer

Move HIP render logic to its dedicated place (#2394)

2023-11-23 13:03:29 -08:00

runtime

bring hip graph back (#2385)

2023-11-24 07:53:44 -08:00

shape

From teeny (#2426)

2023-11-24 12:50:56 -08:00

__init__.py

a lil more beautiful_mnist

2023-11-17 19:53:06 -08:00

graph.py

torch and numpy don't share ops anymore (#2412)

2023-11-23 16:58:10 -08:00

helpers.py

From teeny (#2426)

2023-11-24 12:50:56 -08:00

jit.py

bring hip graph back (#2385)

2023-11-24 07:53:44 -08:00

lazy.py

From teeny (#2426)

2023-11-24 12:50:56 -08:00

mlops.py

add mops to graph, refactor IMAGE (#2100)

2023-10-17 21:27:51 -07:00

ops.py

simple display_name (#2416)

2023-11-23 19:50:23 -08:00

realize.py

don't pass lazybuffer to rawbuffer (#2400)

2023-11-23 09:40:28 -08:00

tensor.py

fix causal mask in Tensor class (#2425)

2023-11-24 18:38:18 -05:00