diff options
Diffstat (limited to 'reference_model/test/model_runner_tests.cpp')
-rw-r--r-- | reference_model/test/model_runner_tests.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/reference_model/test/model_runner_tests.cpp b/reference_model/test/model_runner_tests.cpp index 820ed63..7cf9d68 100644 --- a/reference_model/test/model_runner_tests.cpp +++ b/reference_model/test/model_runner_tests.cpp @@ -111,8 +111,10 @@ TEST_SUITE("model_runner") output.data = reinterpret_cast<uint8_t*>(dstData.data()); output.size = dstData.size() * sizeof(float); + tosa_acc_size_t acc_size = tosa_acc_size_fp32_t; + // Execution - auto status = tosa_run_avg_pool2d(input, kernel, stride, pad, 0, 0, output, {}); + auto status = tosa_run_avg_pool2d(input, kernel, stride, pad, acc_size, 0, 0, output, {}); CHECK((status == tosa_status_valid)); // Compare results |