README.md

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