Commit Graph

512 Commits

Author SHA1 Message Date
Prashant Kumar
f3f0c36902 Add support to pass multiple inputs. 2022-03-22 16:00:01 +00:00
Elias
cc9eea9c1f added functionality to run from c rather than the command line 2022-03-19 00:10:26 +00:00
Prashant Kumar
a89cf2d65f Add examples of shark_inference via torch.script, torch.jit_trace, and
aot.
2022-03-15 21:04:26 +05:30
Elias
2416936ecd added parallel cuda execution 2022-03-15 00:01:57 +00:00
Prashant Kumar
e6115da192 Add support for training in shark_runner.
Added support for training via AOT_module in shark_runner.
2022-03-13 21:50:46 +05:30
Prashant Kumar
fba169f456 Add support of AOT_Module in shark_runner.
Added support of AOT_Module for inference on simple example.
2022-03-12 22:07:30 +05:30
Prashant Kumar
7a0296f359 Added missing set_dep_pypaths scripts. 2022-03-11 22:21:41 +05:30
Prashant Kumar
9f252d688a [S-RUNNER] Initial commit of Shark Runner Inferene API.
Added shark-runner inference API, fully functional given only one input.
2022-03-11 22:21:41 +05:30
Elias
e5174ca097 wip for dynamic indexes 2022-03-10 18:45:09 +00:00
Elias
5223ff69fe removed cmake artifacts (somehow initial commit on this didn't go through) 2022-03-09 19:37:35 +00:00
Elias
3ea6174583 now runs on cuda 2022-03-09 19:33:29 +00:00
elias
d0fd43b897 Initial Commit 2022-03-04 16:51:50 -08:00