From f9c0ceea99e197ab14f779eb51c5e1479dbeb4dd Mon Sep 17 00:00:00 2001
From: Won Jeon <won.jeon@arm.com>
Date: Mon, 18 Sep 2023 16:32:45 -0700
Subject: Add 0-rank tensor support for concat in framework test

Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: Iff77091e4a57f487431ffbf7ac1c89301a153c8b
---
 verif/frameworks/arg_gen.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'verif/frameworks/arg_gen.py')

diff --git a/verif/frameworks/arg_gen.py b/verif/frameworks/arg_gen.py
index a25c205..c385274 100644
--- a/verif/frameworks/arg_gen.py
+++ b/verif/frameworks/arg_gen.py
@@ -45,6 +45,10 @@ class ArgGen:
     @staticmethod
     def agAxes(op, shapes, rng):
         axes = []
+        if shapes == ():
+            axes.append(["_axis_0", [0]])
+            return axes
+
         for i in range(-len(shapes), len(shapes), 1):
             if i >= 0:
                 axes.append(["_axis_{}".format(i), [i]])
-- 
cgit v1.2.1