Commit Graph

1743 Commits

Author SHA1 Message Date
powderluv
cb67b0b3f1 Update README.md 2022-04-25 23:33:30 -07:00
powderluv
58f35f6ae3 Update python-package.yml
add workflow dispatch
2022-04-21 16:35:19 -07:00
powderluv
2d716fd3f6 Create python-package.yml 2022-04-21 16:32:49 -07:00
Prashant Kumar
37468f7bb8 Enable Shark Trainer APIs 2022-04-21 20:31:30 +05:30
Prashant Kumar
d9f47b59f2 Modify setup_venv to incorporate python version. 2022-04-18 17:14:58 +05:30
Prashant Kumar
8aed3ad5e2 Delete README.md since it's redundant. 2022-04-18 11:31:19 +05:30
Prashant Kumar
5991a90284 Modify install.sh and update readme. 2022-04-18 11:29:15 +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
Prashant Kumar
0502c1b384 Add necessary packages installation in install.sh.
I can't find how to maintain order in requirements.txt and
hence install.sh solves this issue.
2022-04-14 22:05:51 +05:30
anush elangovan
e0005b6335 Add github python gitignore 2022-04-13 13:24:38 -07:00
powderluv
27e5ffffb8 Update README.md 2022-04-13 13:19:18 -07:00
anush elangovan
5008ee5f86 Update README 2022-04-13 13:16:49 -07:00
anush elangovan
e3befc0e32 Remove old dshark 2022-04-13 13:15:25 -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
Prashant Kumar
59485f571e Update readme.md of shark_runner
Updated the readme.md of shark_runner and added setup.py for pip
installation.
2022-04-01 01:00:32 +05:30
Prashant Kumar
d11accd6ad Update model tracking. 2022-03-30 19:46:22 +05:30
Ramiro Leal-Cavazos
7d429e3c92 Update torch-mlir utils + remove unecessary imports in shark_runner
This commit updates the `torch_mlir_utils.py` to use the new
torch-mlir python package structure. This commit also removes
unecessary imports from the Python scripts in the shark_runner directory.
2022-03-30 19:42:19 +05:30
Prashant Kumar
d5a626a4c6 Changing README.md. 2022-03-30 19:37:31 +05:30
Prashant Kumar
f30d24ab2b Update README.md with model tracking information
Hugging face models and TorchVision models tracking sheet has been added.
2022-03-30 19:33:38 +05:30
Prashant Kumar
1cdc9cd198 Add bert aot forward example.
Add option to attach custom inference function to aot_module.
Also, added an example of bert inference to run via aot.
2022-03-29 21:41:18 +05:30
Eliasj42
29113c4b71 Merge pull request #9 from NodLabs/rest-api-rework
fixed segfaults when running iree
2022-03-25 14:30:17 -07:00
Prashant Kumar
77a2525ace Fix bug related to aot module.
Single element is returned as is and tuple is not created for that.
2022-03-25 18:32:10 +05:30
Elias
6a2efa35a5 fixed segfaults when running iree 2022-03-23 20:59:23 +00:00
Eliasj42
0e762fabbd Merge pull request #8 from NodLabs/rest-api-rework
reworked how the server and client were structured
2022-03-22 19:47:42 -07:00
Elias
d447d874dd reworked how the server and client were structured so that they could be more usable 2022-03-22 21:42:36 +00:00
powderluv
cf2e1bc4f0 Merge pull request #7 from pashu123/mulinps
Add support to pass multiple inputs.
2022-03-22 10:10:50 -07:00
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