chenyu 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

..

__init__.py

distributed world (#1481)

2023-08-10 10:00:51 -07:00

collectives.py

train cifar using multigpu (#1529)

2023-08-18 09:35:44 -07:00

world.py

Symbolic Shape JIT main PR (#1353)

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