Commit Graph

55 Commits

Author SHA1 Message Date
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
Prashant Kumar
37468f7bb8 Enable Shark Trainer APIs 2022-04-21 20:31:30 +05:30
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
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