max
1d2c3ee584
Demo for how dynamo with mlir backend can exist out of (dynamo) tree.
2022-06-01 18:57:58 -05:00
Maksim Levental
05dffd4d59
Demo for how dynamo with mlir backend can exist out of (dynamo) tree. ( #78 )
2022-06-01 14:57:04 -07:00
powderluv
27a33e636d
Update nightly.yml
...
Remove spurious test package
2022-06-01 14:44:39 -07:00
max
4572d2bfba
Demo for how dynamo with mlir backend can exist out of (dynamo) tree.
2022-06-01 15:32:17 -05:00
Stanley Winata
b271daa748
Rectify Vulkan Flags and Command to obtain Vulkan Target Triple ( #77 )
...
-Add correct vulkan flag to run minilm_jit
-Add automatic specification of target triple for better speed.
-Will work when https://github.com/nod-ai/SHARK-Runtime/releases/tag/candidate-20220601.20 gets
released, for now we can replace the shark release link to this on setup_venv.sh to work.
2022-06-01 11:33:53 -07:00
Maksim Levental
5530b56921
Shark Eager Mode jupyter notebook ( #76 )
2022-06-01 09:21:32 -07:00
Ean Garvey
87098e315f
Pass --iree-hal-cuda-llvm-target-arch for GPU execution automatically. ( #66 )
...
* Pass --iree-hal-cuda-llvm-target-arch for GPU execution automatically.
Add IREE CUDA compute capability flag only if sm_70 or later.
* Add Validate Models step to nightly workflow.
2022-06-01 08:38:25 -07:00
Maksim Levental
f57730d2db
Fix type ( #75 )
2022-06-01 08:30:12 -07:00
Vivek Khandelwal
d98d6a065d
Update README.md ( #74 )
2022-06-01 06:53:05 -07:00
Vivek Khandelwal
6081689c9f
Add test for Resnet101 and remove Vgg11 test ( #73 )
2022-06-01 00:49:07 -07:00
Prashant Kumar
04d23e504b
Modify vgg16 to vgg11 to ignore memory limit error.
2022-05-27 21:57:40 +05:30
Prashant Kumar
04291fdfcb
Add vgg and alexnet models to passing tests.
2022-05-27 21:35:02 +05:30
Prashant Kumar
cee02f6a61
Add frontend error checks.
2022-05-27 15:45:45 +05:30
Prashant Kumar
c69baa3b1e
Rename and refactoring.
2022-05-27 15:17:08 +05:30
Prashant Kumar
8294dc3f20
Bug fix related to torch-shark trainer.
2022-05-27 14:50:28 +05:30
Prashant Kumar
e5517f63f5
Add the refactored training module.
2022-05-27 14:21:34 +05:30
powderluv
96acb32fc1
Update test-models.yml
2022-05-26 21:06:20 -07:00
powderluv
a72f727501
Update test-models.yml
2022-05-26 21:03:35 -07:00
Stanley Winata
7b4c7f9bde
Merge pull request #62 from nod-ai/benchmark
...
Add Shark Benchmark
2022-05-26 17:18:55 -07:00
stanley
91867e1fc2
Add Shark Benchmark
...
-Introduce SharkBenchmark that bench models on regular torch, shark-py, and shark-c.
-Integrate iree-benchmark-module into Shark.
2022-05-27 00:07:31 +00:00
Prashant Kumar
18a4423026
Clearing the shark_trainer API for future rework.
2022-05-26 20:15:18 +05:30
Prashant Kumar
15c99adab8
Refactor the whole SharkInference API.
...
1. SharkInference API is refactored to target any frontends or files,
i.e. pytorch, tensorflow, mhlo, linalg, tosa.
2. Refactor all torch APIs.
2022-05-26 18:55:47 +05:30
Eliasj42
ad16e54d4e
updated README ( #61 )
2022-05-25 21:51:02 -07:00
Maksim Levental
ef6f3a7b74
Eager mode with IREE ( #25 )
...
Eager mode with IREE
reenable iree integration
2022-05-25 21:49:19 -07:00
Ean Garvey
772f60c313
Add support for TensorFlow modules + TF miniLM example ( #51 )
2022-05-25 18:50:43 -07:00
Prashant Kumar
737be5be09
Remove AOTModule and update vulkan iree args.
2022-05-25 21:17:17 +05:30
Stanley Winata
c8fcfd6077
Set parallel work as manually configured option ( #58 )
2022-05-24 15:26:28 -07:00
Stanley Winata
5ece8f6f5d
Fix Test Issues on macos ( #57 )
...
-Added a separate pytest config for macos
-Adjusted shark parser to ignore unknown args
2022-05-24 14:40:35 -07:00
Prashant Kumar
6a358cb90a
Update the examples removing functorch dependency.
2022-05-24 19:10:01 +05:30
Prashant Kumar
96577000a3
Remove functorch dependency.
2022-05-24 18:42:35 +05:30
Prashant Kumar
126771fa55
Add gpu tested models.
2022-05-24 11:27:34 +05:30
Prashant Kumar
4b63e0e04a
Add run_on_refbackend and gpu configs. ( #52 )
...
`run_on_refbackend` is added to run linalg_on_tensors_backend compiled
code on torch-mlir's refbackend. Also, added gpu configs and flags.
2022-05-23 14:18:11 -07:00
powderluv
2df9128749
Update README.md
2022-05-23 14:12:55 -07:00
powderluv
c854bfc313
Update nightly.yml
...
remove snapshot name
2022-05-17 12:15:01 -07:00
powderluv
d18adad3e7
Rename python-package.yml to nightly.yml
2022-05-17 08:36:56 -07:00
Prashant Kumar
96c28685ce
Added device map and runs test according to the driver.
...
Automatically runs test according to the device driver present.
2022-05-17 19:33:54 +05:30
Prashant Kumar
f20fa5de3b
Update README.md
2022-05-17 19:26:39 +05:30
Prashant Kumar
0bd5f22f63
Remove redundant examples.
2022-05-17 17:50:16 +05:30
Prashant Kumar
63f9b206ac
Modify pytest to run a certain test.
2022-05-17 16:58:08 +05:30
powderluv
3eb222fbdd
Update python-package.yml
2022-05-17 00:00:59 -07:00
powderluv
e3a7c8046e
Update python-package.yml
2022-05-16 23:56:15 -07:00
powderluv
19a48d86b8
Update python-package.yml
2022-05-16 23:52:32 -07:00
Prashant Kumar
dcc1bbac14
Add pytests-parametrize with device and dynamic option.
...
Added pytests-parametrize with device and dynamic options, also
added worker flag in pytest to run tests concurrently.
2022-05-17 11:07:56 +05:30
Prashant Kumar
979604a66e
Fix minor bug.
2022-05-16 22:51:22 +05:30
Prashant Kumar
84d0233a3f
Shark-runtime repository update.
2022-05-16 17:46:24 +05:30
anush elangovan
49bec49c53
prep for pypi
2022-05-16 05:44:18 +00:00
anush elangovan
628d5aa112
Ingore inference tests
2022-05-16 03:04:51 +00:00
Prashant Kumar
a77f4d8acb
Update README.md
2022-05-16 08:18:24 +05:30
powderluv
eccc5b1900
Update README.md
2022-05-15 15:48:45 -07:00
Stanley Winata
e2d5a49dc6
Merge pull request #45 from raikonenfnu/UpdateVenv
...
Install package to pip venv during setup_venv
2022-05-15 15:28:16 -07:00
