chenyuandGitHub ae39cf84ab Symbolic Shape JIT main PR (#1353) ...

* Symbolic Shape JIT

update tests

2 variables symbolic ops, adding more tests

test passing

cleanup

* more test cases

* single flag

* review update

* jit attention one piece

* realize

* symbolic_jit test for cuda

* old artifact

* works with cuda gpu but failed ci

* CUDACPU

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

..

codegen

PTX Reintegration and Passing Tests (#1512)

2023-08-16 16:20:20 -07:00

nn

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

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

renderer

symbolic codegen and exec (#1552)

2023-08-16 14:43:41 -07:00

runtime

Symbolic Shape JIT main PR (#1353)

2023-08-18 14:39:55 -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

Symbolic Shape JIT main PR (#1353)

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

jit.py

Symbolic Shape JIT main PR (#1353)

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

lazy.py

support for p2p buffer transfers (#1523)

2023-08-14 22:39:57 -07:00

mlops.py

cleanup mlops (#1521)

2023-08-10 19:53:28 -07:00

ops.py

Symbolic Shape JIT main PR (#1353)

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

state.py

add stable diffusion and llama (#1471)

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

tensor.py

Gather bugfix (#1561)

2023-08-16 19:53:14 -04:00