From 28e0919321142d75c8a08234cf760c3c2550f830 Mon Sep 17 00:00:00 2001
From: PhaneeshB <b.phaneesh@gmail.com>
Date: Fri, 23 Jun 2023 18:31:38 +0530
Subject: [PATCH] Add AMD cpu device

---
 shark/iree_utils/_common.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/shark/iree_utils/_common.py b/shark/iree_utils/_common.py
index 8c79243129..3fc689fbe7 100644
--- a/shark/iree_utils/_common.py
+++ b/shark/iree_utils/_common.py
@@ -63,6 +63,7 @@ def get_supported_device_list():
 _IREE_DEVICE_MAP = {
     "cpu": "local-task",
     "cpu-task": "local-task",
+    "AMD-AIE": "local-task",
     "cpu-sync": "local-sync",
     "cuda": "cuda",
     "vulkan": "vulkan",
@@ -81,6 +82,7 @@ def iree_target_map(device):
 _IREE_TARGET_MAP = {
     "cpu": "llvm-cpu",
     "cpu-task": "llvm-cpu",
+    "AMD-AIE": "llvm-cpu",
     "cpu-sync": "llvm-cpu",
     "cuda": "cuda",
     "vulkan": "vulkan",