George Hotz 228f30b96a multitensor jit (#3149) ...

* initial multitensor jit support and tests

* Added graphs to multitensor jit and updated tests

* update unbind api

* fix set device, add TinyJit to resnet

* update_stats includes device

---------

Co-authored-by: ramenguy99 <ramenguy99@gmail.com>

2024-01-16 09:09:15 -08:00

4.7 KiB

Raw Blame History

View Raw