minor change

r-abishek · Sep 27, 2023 · 9f6b6d4 · 9f6b6d4
1 parent c413836
commit 9f6b6d4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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())