AMD-SHARK-Studio/shark/sharkdynamo at enable_caching_alloc - AMD-SHARK-Studio - AtHeartEngineering

github/AMD-SHARK-Studio

mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-02-19 11:56:43 -05:00

Files

History

Prashant Kumar 3595b4aaff Incorporate latest changes in the shark_dynamo backend.

2023-02-08 20:37:30 +05:30

..

__init__.py

Add sharkdynamo which combines shark with torchdynamo.

2022-09-23 22:40:02 +05:30

README.md

Add sharkdynamo which combines shark with torchdynamo.

2022-09-23 22:40:02 +05:30

utils.py

Incorporate latest changes in the shark_dynamo backend.

2023-02-08 20:37:30 +05:30

README.md

Install torchdynamo
- git clone https://github.com/pytorch/torchdynamo.git
- cd torchdynamo
- python -m pip install -r requirements.txt
- python setup.py develop
Install functorch
- python -m pip install -v "git+https://github.com/pytorch/pytorch.git@$(python -c "import torch.version; print(torch.version.git_version)")#subdirectory=functorch"
Run examples.
- python shark/examples/shark_dynamo/basic_examples.py