search

nod-ai / iree-amd-aie

IREE plugin repository for the AMD AIE accelerator
Apache License 2.0
69 stars 30 forks source link

[Test] Update run.py to run only `target_device` specific tests #887

Closed Abhishek-Varma closed 1 week ago

Abhishek-Varma commented 1 week ago

-- This commit updates run.py to create test instances specifying the intended target device(s) to run them on. And accordingly run only those test instances whose intended target device list matches the supplied command-line arg target_device.

Signed-off-by: Abhishek Varma abhvarma@amd.com

Keeping this PR logically separate from the Strix ukernel one - helps compartmentalizing.

Abhishek-Varma commented 1 week ago

Looks like a good approach to me.

I think the code will be simpler if target_device is a member of config.

I didn't know about ABC... interesting how python has added things to be more like C++!

Good catch. target_device was already a member of config. So have removed that from the function declaration/invocation now.