MLBEDSW-3465: Add memory settings into sys config
Signed-off-by: Diqing Zhong <diqing.zhong@arm.com>
Change-Id: I4a5c53d0c5957595fc639b174b2b227ea043d409
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index de97710..257cb5f 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -19,6 +19,7 @@
import enum
import uuid
from collections import defaultdict
+from enum import auto
from functools import lru_cache
from typing import Dict
from typing import List
@@ -62,6 +63,22 @@
return self.name
+class BandwidthDirection(enum.IntEnum):
+ Read = 0
+ Write = auto()
+ Size = auto()
+
+ def display_name(self):
+ return self.name
+
+ def identifier_name(self):
+ return self.name.lower()
+
+ @staticmethod
+ def all():
+ return (BandwidthDirection.Read, BandwidthDirection.Write)
+
+
class MemArea(enum.IntFlag):
Unknown = 0
Sram = 1