David HouandGitHub 5cfcc2a8d7 support MLB reshaping on-axis for evenly sharded (#3484) ...

* support MLB reshaping on-axis for evenly sharded

* update test

* not -> !=

2024-02-23 07:51:36 -05:00

..

codegen

Revert "wmma: add CUDA tensor core (#3464)" (#3474)

2024-02-22 11:58:16 +01:00

features

support MLB reshaping on-axis for evenly sharded (#3484)

2024-02-23 07:51:36 -05:00

nn

remove cpu and torch backends (#3399)

2024-02-15 16:55:39 +01:00

renderer

Revert "wmma: add CUDA tensor core (#3464)" (#3474)

2024-02-22 11:58:16 +01:00

runtime

Revert "wmma: add CUDA tensor core (#3464)" (#3474)

2024-02-22 11:58:16 +01:00

shape

symbolic use mod for rmod and use floordiv for rfloordiv (#3485)

2024-02-23 01:05:13 -05:00

__init__.py

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

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

device.py

uops flop counter (#3373)

2024-02-20 09:36:30 +01:00

dtype.py

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

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

helpers.py

bump db version to 11 (#3398)

2024-02-14 10:13:18 -05:00

lazy.py

_reduce_op is axis based now (#3462)

2024-02-21 16:36:31 +01:00

mlops.py

change frontend reduce API to use axis (#3460)

2024-02-21 12:26:17 +01:00

ops.py

_reduce_op is axis based now (#3462)

2024-02-21 16:36:31 +01:00

realize.py

disk runner (prereq for interpreted removal) (#3421)

2024-02-15 18:14:05 +01:00

tensor.py

fix softmax and log_softmax for 0d tensor (#3463)

2024-02-21 11:30:30 -05:00