Skip to content
Commit 49c51ad8 authored by Justin Ngo's avatar Justin Ngo Committed by Eric Kunze
Browse files

Add Torch Unit Test Suite to Reference Model



Unit test test suite for PyTorch utilizing FxImporter to lower PyTorch
graphs to Torch-MLIR then to TOSA IR. Current infrastructure is mostly
based on the existing TF/TFL frameworks test suite.

This is the initial work for the Torch unit test suite. The end goal
is to add more Torch unit tests for a more robust test suite.

Signed-off-by: default avatarJustin Ngo <justin.ngo@arm.com>
Change-Id: I829ddf708556488166bc2bfdd4b52a1cfb4fc34d
parent c7c8aa3d
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment