618 Commits

Author SHA1 Message Date
Prashant Kumar
f590bd56c3 Add bert_training example. 2022-05-06 21:57:46 +05:30
Prashant Kumar
b4114c0938 Make backward API workable. 2022-05-06 14:51:44 +05:30
Prashant Kumar
08df345fbc Add make_fx whole graph extract. 2022-05-04 20:58:59 +05:30
Vivek Khandelwal
7f6159462d Add num_iterations and num_warmup_iterations args. 2022-05-04 20:38:19 +05:30
Prashant Kumar
db5be15310 Centralize the parser location. Also add the --device flag.
Centralized the shark_args parser. Also added the --device flag that
specifies the device on which the inference or training is to be done.
2022-05-02 15:35:22 +05:30
Stanley Winata
b8602d0b64 Automatic selection of Target Triple from Sytem clang (#29)
-To enable running of models and further optimization on CPU that is not x86
2022-04-29 15:49:37 -07:00
Prashant Kumar
bd212634c1 Add CI for testing of various models. (#28)
Added pytest CI for testing for Pytorch Inference and vision models.
2022-04-27 06:52:49 -07:00
Prashant Kumar
3ed5a9c84e Change the python import of shark. 2022-04-26 20:55:33 +05:30
Prashant Kumar
37468f7bb8 Enable Shark Trainer APIs 2022-04-21 20:31:30 +05:30
Prashant Kumar
8aed3ad5e2 Delete README.md since it's redundant. 2022-04-18 11:31:19 +05:30
powderluv
8ab38a3ca0 Merge pull request #19 from NodLabs/ean-export-modules
Add shark-runner command line options for saving modules.
2022-04-15 16:17:04 -07:00
monorimet
1f758a4278 Add in option to specify path to which module files are saved. 2022-04-15 18:11:56 -05:00
monorimet
692cd180f6 Add shark-runner command line options for saving modules as .mlir or .vmfb 2022-04-15 01:10:30 -05:00
anush elangovan
e0005b6335 Add github python gitignore 2022-04-13 13:24:38 -07:00
Prashant Kumar
c586564356 [shark] Update readme.
Update readme to point to the shark location.
2022-04-13 21:53:48 +05:30
Prashant Kumar
dcb5e16b20 Update README.md 2022-04-04 21:29:32 +05:30
Prashant Kumar
33fe8f9c19 Remove unnecessary files. 2022-04-04 21:27:58 +05:30
Prashant Kumar
f9864b4ce1 Modify the directory structure to remove conflicts.
The directory structure is modified, also disabled the aot_module
unless the shark_trainer is deployed.
2022-04-04 21:23:33 +05:30