diff --git a/PyITA/ITA.py b/PyITA/ITA.py index 4b7f01e..24f7b0b 100644 --- a/PyITA/ITA.py +++ b/PyITA/ITA.py @@ -1007,7 +1007,10 @@ def calculate_tensor_stats(tensor, name, tol = 1e-1): print(f"WARNING: {name} is constant!") print(f"{name} Mean-Squared Difference (row) : {similarity_row:5.1f}") print(f"{name} Mean-Squared Difference (column): {similarity_column:5.1f}") - raise ValueError(f"Tensor {name} is constant! This is a bad test vector!") + if kwargs['skip_vector_validation'] is False: + raise ValueError(f"Tensor {name} is constant! This is a bad test vector!") + else: + print(f" WARNING: Tensor {name} is constant! This is a bad test vector!") else: print(" WARNING: Tensor is constant!") print(f" Mean-Squared Difference (row) : {similarity_row:5.1f}") diff --git a/testGenerator.py b/testGenerator.py index 7082002..56ce83e 100644 --- a/testGenerator.py +++ b/testGenerator.py @@ -106,6 +106,7 @@ class ArgumentDefaultMetavarTypeFormatter(argparse.ArgumentDefaultsHelpFormatter action = 'store_true', help = 'Disable partial softmax calculation') self.group1.add_argument('--no-bias', default = False, action = 'store_true', help = 'Disable bias') + self.group1.add_argument('--skip-vector-validation', action = 'store_true', help = 'Skip validation of test vectors') self.group1.add_argument('--export-snitch-cluster', action = 'store_true', help = 'Export for snitch cluster') self.group1.add_argument('--export-mempool', action = 'store_true', help = 'Export for mempool') self.group1.add_argument('--export-rom', action = 'store_true', help = 'Export ROM configuration')