Function TEST

dali/pipeline/executor/executor2/stream_assignment_test.cc:136–149

}  // namespace

TEST(Exec2Test, StreamAssignment_Single_OnlyCPU) {
  graph::OpGraph::Builder b;
  b.Add("a",
        SpecCPU()
        .AddOutput("a->out", StorageDevice::CPU));
  b.AddOutput("a->out_cpu");
  auto g = std::move(b).GetGraph(true);
  ExecGraph eg;
  eg.Lower(g);

  StreamAssignment<StreamPolicy::Single> assignment(eg);
  auto map = MakeNodeMap(eg);
  EXPECT_EQ(assignment[map["a"]], std::nullopt);
}

TEST(Exec2Test, StreamAssignment_Single_CPUMixedGPU) {
  graph::OpGraph::Builder b;