diff options
author | Joel Liang <joel.liang@arm.com> | 2017-12-28 10:09:51 +0800 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:42:33 +0000 |
commit | 1c5ffd612979d40730feecc4f43fa6c9f177f2e3 (patch) | |
tree | 510367ba70fcad4374d2cf9cdccdb3bc715c216c /tests/networks | |
parent | c5a7e59655b61ad617fa34a4fb00e1a007c8255a (diff) | |
download | ComputeLibrary-1c5ffd612979d40730feecc4f43fa6c9f177f2e3.tar.gz |
APPBROWSER-359: Sync tensor in GC benchmark tests
Change-Id: I22c1aa92e70d6143bbcec90e9e7de9f1ce1c1e55
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/114635
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Diffstat (limited to 'tests/networks')
-rw-r--r-- | tests/networks/AlexNetNetwork.h | 7 | ||||
-rw-r--r-- | tests/networks/LeNet5Network.h | 7 | ||||
-rw-r--r-- | tests/networks/MobileNetNetwork.h | 7 | ||||
-rw-r--r-- | tests/networks/MobileNetV1Network.h | 7 |
4 files changed, 28 insertions, 0 deletions
diff --git a/tests/networks/AlexNetNetwork.h b/tests/networks/AlexNetNetwork.h index 448cf31914..819111f897 100644 --- a/tests/networks/AlexNetNetwork.h +++ b/tests/networks/AlexNetNetwork.h @@ -564,6 +564,13 @@ public: smx.run(); } + /** Sync the results */ + void sync() + { + sync_if_necessary<TensorType>(); + sync_tensor_if_necessary<TensorType>(output); + } + private: struct DirectConv { diff --git a/tests/networks/LeNet5Network.h b/tests/networks/LeNet5Network.h index f8aac30807..a46489f88c 100644 --- a/tests/networks/LeNet5Network.h +++ b/tests/networks/LeNet5Network.h @@ -230,6 +230,13 @@ public: smx.run(); } + /** Sync the results */ + void sync() + { + sync_if_necessary<TensorType>(); + sync_tensor_if_necessary<TensorType>(output); + } + private: unsigned int _batches{ 0 }; diff --git a/tests/networks/MobileNetNetwork.h b/tests/networks/MobileNetNetwork.h index 1bc8ad9a0c..8c3cb1fb2d 100644 --- a/tests/networks/MobileNetNetwork.h +++ b/tests/networks/MobileNetNetwork.h @@ -251,6 +251,13 @@ public: reshape.run(); } + /** Sync the results */ + void sync() + { + sync_if_necessary<TensorType>(); + sync_tensor_if_necessary<TensorType>(output); + } + private: void depthwise_conv_block_init(unsigned int idx, unsigned int ifm, unsigned int ofm) { diff --git a/tests/networks/MobileNetV1Network.h b/tests/networks/MobileNetV1Network.h index dbe3f491f0..0957c6b555 100644 --- a/tests/networks/MobileNetV1Network.h +++ b/tests/networks/MobileNetV1Network.h @@ -300,6 +300,13 @@ public: smx.run(); } + /** Sync the results */ + void sync() + { + sync_if_necessary<TensorType>(); + sync_tensor_if_necessary<TensorType>(output); + } + private: void depthwise_conv_block_init(unsigned int idx, unsigned int ifm, unsigned int ofm) { |