diff options
Diffstat (limited to 'src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp')
| -rw-r--r-- | src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp b/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp index 125b7e62b1..298901e640 100644 --- a/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp +++ b/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp @@ -41,8 +41,9 @@ LayerTestResult<T, 2> SimpleFullyConnectedTestImpl( outputHandle->Allocate(); CopyDataToITensorHandle(inputHandle.get(), &input[0][0][0][0]); - workloadFactory.Finalize(); + workloadFactory.Acquire(); workload->Execute(); + workloadFactory.Release(); CopyDataFromITensorHandle(&result.output[0][0], outputHandle.get()); |