ArmNN
 21.08
ExecutionFrameTest.cpp File Reference
#include <doctest/doctest.h>
#include <ExecutionFrame.hpp>

Go to the source code of this file.

Functions

 TEST_SUITE ("NextExecutionFrameTestSuite")
 

Function Documentation

◆ TEST_SUITE()

TEST_SUITE ( "NextExecutionFrameTestSuite"  )

Definition at line 17 of file ExecutionFrameTest.cpp.

References ExecutionFrame::ExecuteWorkloads(), and ExecutionFrame::SetNextExecutionFrame().

18 {
19 TEST_CASE("NextExecutionFrameTest")
20 {
21  armnn::ExecutionFrame executionFrameA;
22  armnn::ExecutionFrame executionFrameB;
23  armnn::ExecutionFrame executionFrameC;
24 
25  executionFrameA.SetNextExecutionFrame(&executionFrameB);
26  executionFrameB.SetNextExecutionFrame(&executionFrameC);
27  //not setting C to check that the default setting is nullptr.
28 
29  auto nextExecutionFrameA = executionFrameA.ExecuteWorkloads(nullptr);
30  auto nextExecutionFrameB = executionFrameB.ExecuteWorkloads(&executionFrameA);
31  auto nextExecutionFrameC = executionFrameC.ExecuteWorkloads(&executionFrameB);
32 
33  CHECK_EQ(nextExecutionFrameA, &executionFrameB);
34  CHECK_EQ(nextExecutionFrameB, &executionFrameC);
35 
36  CHECK(!nextExecutionFrameC);
37 
38  CHECK_NE(nextExecutionFrameA, &executionFrameC);
39 }
40 }
IExecutionFrame * ExecuteWorkloads(IExecutionFrame *previousFrame) override
void SetNextExecutionFrame(IExecutionFrame *nextExecutionFrame)