Skip to content
This repository has been archived by the owner on Oct 21, 2023. It is now read-only.

micronDLA/pytorch_mdla

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Adding MDLA to pytorch backend

This folder contains example implementation of mdla backend for pytorch torchdynamo.

Install

Install torchdynamo

Build and Test tmdla

Add api.h, thvector.h, thnets.h and thnets.def into this folder

Then run:

python3 setup.py install

Run examples

Run torch.trace graph with mdla

python3 test.py

Run torchdynamo with mdla

python3 test_dynamo.py

Run a torchvision model with mdla

python3 test_model.py alexnet

Torchscript MDLA

For running MDLA with torchscript refer to here

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published