diff options
Diffstat (limited to 'tests/benchmark/fixtures/DepthwiseSeparableConvolutionLayerFixture.h')
-rw-r--r-- | tests/benchmark/fixtures/DepthwiseSeparableConvolutionLayerFixture.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tests/benchmark/fixtures/DepthwiseSeparableConvolutionLayerFixture.h b/tests/benchmark/fixtures/DepthwiseSeparableConvolutionLayerFixture.h index fa1544042f..c841996c6c 100644 --- a/tests/benchmark/fixtures/DepthwiseSeparableConvolutionLayerFixture.h +++ b/tests/benchmark/fixtures/DepthwiseSeparableConvolutionLayerFixture.h @@ -83,6 +83,12 @@ public: depth_sep_conv_layer.run(); } + void sync() + { + sync_if_necessary<TensorType>(); + sync_tensor_if_necessary<TensorType>(dst); + } + void teardown() { src.allocator()->free(); |