From 9f6b6d465dc2054aa28726df3452360e5088cbd8 Mon Sep 17 00:00:00 2001 From: sampath1117 Date: Wed, 27 Sep 2023 12:01:07 +0000 Subject: [PATCH] minor change --- utilities/test_suite/HOST/Tensor_host_audio.cpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/utilities/test_suite/HOST/Tensor_host_audio.cpp b/utilities/test_suite/HOST/Tensor_host_audio.cpp index 06378c870..a3b760ff1 100644 --- a/utilities/test_suite/HOST/Tensor_host_audio.cpp +++ b/utilities/test_suite/HOST/Tensor_host_audio.cpp @@ -41,6 +41,12 @@ int main(int argc, char **argv) int batchSize = atoi(argv[6]); char *dst = argv[7]; + if (testType == 0 && batchSize != 8) + { + cout << "Error! QA Mode only runs with batchsize 8" << endl; + return -1; + } + // Set case names string funcName = audioAugmentationMap[testCase]; if (funcName.empty())