Add allowed profiles for each extension

Each extension can contain a list of profiles that it works with.

Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: I7cf2ad0c57fc8a36500212b7d554a8153fc2efe5

1 files changed, 4 insertions, 2 deletions

diff --git a/tools/tosa.py b/tools/tosa.py
index e70b297..016aba6 100644
--- a/tools/tosa.py
+++ b/tools/tosa.py
@@ -41,10 +41,11 @@ class TOSAProfile:
 
 
 class TOSAProfileExtension:
-    def __init__(self, name, description, status):
+    def __init__(self, name, description, status, profiles):
         self.name = name
         self.description = description
         self.status = status
+        self.profiles = profiles
         self.ops = []
 
 
@@ -155,7 +156,8 @@ class TOSASpec:
         name = ext.get("name")
         description = ext.get("description")
         status = ext.get("status")
-        return TOSAProfileExtension(name, description, status)
+        profiles = [x.text for x in ext]
+        return TOSAProfileExtension(name, description, status, profiles)
 
     def __load_level(self, level):
         name = level.get("name")