Umut ZenginandGitHub 35bf21276f Argmax/Argmin Feature (#1576) ...

* implemented argmax and argmin

* lint

* lint

* match torch behaviour

* format

* removed flip

2023-08-20 18:46:46 -07:00

..

codegen

not worth the speed (#1584)

2023-08-20 10:24:58 -07:00

nn

Add assert for wrong matmul/dot shapes (#1438)

2023-08-04 18:16:56 -04:00

renderer

fix __launch_bounds__ and benchmark TC MATMUL (#1575)

2023-08-19 10:54:39 -07:00

runtime

fix __launch_bounds__ and benchmark TC MATMUL (#1575)

2023-08-19 10:54:39 -07:00

shape

more symbolic symbolic ops (#1564)

2023-08-18 09:21:41 -07:00

graph.py

Add WHERE ternary (or trinary?) op (#1196)

2023-07-16 00:31:55 -07:00

helpers.py

teeny changes (#1589)

2023-08-20 13:38:38 -07:00

jit.py

Symbolic Shape JIT main PR (#1353)

2023-08-18 14:39:55 -07:00

lazy.py

teeny changes (#1589)

2023-08-20 13:38:38 -07:00

mlops.py

teeny changes (#1589)

2023-08-20 13:38:38 -07:00

ops.py

not worth the speed (#1584)

2023-08-20 10:24:58 -07:00

state.py

add stable diffusion and llama (#1471)

2023-08-06 21:31:51 -07:00

tensor.py

Argmax/Argmin Feature (#1576)

2023-08-20 18:46:46 -07:00