diff options
Diffstat (limited to 'app/test-mldev/test_inference_interleave.c')
-rw-r--r-- | app/test-mldev/test_inference_interleave.c | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/app/test-mldev/test_inference_interleave.c b/app/test-mldev/test_inference_interleave.c index 23b8efe4f0..9503dda042 100644 --- a/app/test-mldev/test_inference_interleave.c +++ b/app/test-mldev/test_inference_interleave.c @@ -7,6 +7,7 @@ #include "ml_common.h" #include "test_inference_common.h" +#include "test_stats.h" static int test_inference_interleave_driver(struct ml_test *test, struct ml_options *opt) @@ -58,7 +59,8 @@ test_inference_interleave_driver(struct ml_test *test, struct ml_options *opt) ml_inference_iomem_destroy(test, opt, fid); } - ml_inference_throughput_get(test, opt); + for (fid = 0; fid < opt->nb_filelist; fid++) + ml_stats_get(test, opt, RTE_ML_DEV_XSTATS_MODEL, fid); for (fid = 0; fid < opt->nb_filelist; fid++) { ret = ml_model_stop(test, opt, &t->model[fid], fid); @@ -70,6 +72,8 @@ test_inference_interleave_driver(struct ml_test *test, struct ml_options *opt) goto error; } + ml_stats_get(test, opt, RTE_ML_DEV_XSTATS_DEVICE, -1); + ml_throughput_get(test, opt); ml_inference_mem_destroy(test, opt); ret = ml_inference_mldev_destroy(test, opt); |