From f890e92c2ece805d6f23dcb4954d5285c2656124 Mon Sep 17 00:00:00 2001
From: Arne Symons <arnesymons@hotmail.com>
Date: Thu, 16 May 2024 11:34:53 +0200
Subject: [PATCH] add resnet18 first layer name entry to tpu_like mapping

---
 inputs/mapping/tpu_like.yaml  | 16 ++++++++++++++++
 inputs/workload/resnet18.yaml |  4 ++--
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/inputs/mapping/tpu_like.yaml b/inputs/mapping/tpu_like.yaml
index 9528b97f..185bbeb6 100644
--- a/inputs/mapping/tpu_like.yaml
+++ b/inputs/mapping/tpu_like.yaml
@@ -1,3 +1,4 @@
+# Default entry in case layer name and operator type are not available
 - name: default
   core_allocation: [1]
   spatial_mapping:
@@ -10,6 +11,20 @@
     W: I2
     I: I1
 
+# Mapping entry for specific layer name
+- name: example_name_of_layer0
+  core_allocation: [1]
+  spatial_mapping:
+    D1:
+      - K, 32
+    D2:
+      - C, 32
+  memory_operand_links:
+    O: O
+    W: I2
+    I: I1
+
+# Entry for 'Add' operator type
 - name: Add
   core_allocation: [1]
   spatial_mapping:
@@ -22,6 +37,7 @@
     W: I2
     I: I1
 
+# Entry for 'Pooling' operator type
 - name: Pooling
   core_allocation: [1]
   spatial_mapping:
diff --git a/inputs/workload/resnet18.yaml b/inputs/workload/resnet18.yaml
index 9eab61ae..f5d89a33 100644
--- a/inputs/workload/resnet18.yaml
+++ b/inputs/workload/resnet18.yaml
@@ -1,6 +1,6 @@
 - id: 0 # conv1 stride 2
-  name: example_name_of_layer0
-  operator_type: Conv
+  name: example_name_of_layer0  # name can be used to specify mapping
+  operator_type: Conv  # operator_type can be used to specify mapping
   equation: O[b][k][oy][ox]+=W[k][c][fy][fx]*I[b][c][iy][ix]
   dimension_relations: [ix=2*ox+1*fx, iy=2*oy+1*fy]
   loop_dims: [B, K, C, OY, OX, FY, FX]