From 6f57e6e665094959aed40c0e388ac81fbd118720 Mon Sep 17 00:00:00 2001 From: Jeremy Johnson Date: Tue, 30 Jan 2024 16:10:50 +0000 Subject: Main Compliance: RFFT2D support Correct ref model to produce imaginery values of zero as specification indicates at certain output positions. Fix up precise and abs modes for RFFT2D in ref model to produce correct results and bounds using abs weights. Signed-off-by: Jeremy Johnson Change-Id: I33767e4219a260278f7933f28b1799223a95a3cc --- verif/generator/tosa_arg_gen.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) (limited to 'verif/generator/tosa_arg_gen.py') diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py index f6a46b4..a4bced3 100644 --- a/verif/generator/tosa_arg_gen.py +++ b/verif/generator/tosa_arg_gen.py @@ -2821,6 +2821,31 @@ class TosaArgGen: # Return list of tuples: (arg_str, args_dict) return arg_list + @staticmethod + def agRFFT2d(testGen, opName, shapeList, dtype, error_name=None): + arg_list = [] + + shape = shapeList[0] + dot_products = gtu.product(shape) + ks = shape[1] * shape[2] # H*W + args_dict = { + "dot_products": dot_products, + "shape": shape, + "ks": ks, + "acc_type": dtype, + } + arg_list.append(("", args_dict)) + + arg_list = TosaArgGen._add_data_generators( + testGen, + opName, + dtype, + arg_list, + error_name, + ) + # Return list of tuples: (arg_str, args_dict) + return arg_list + # Helper function for reshape. Gets some factors of a larger number. @staticmethod def getFactors(val, start=1): -- cgit v1.2.1