// Copyright (c) 2018 Graphcore Ltd. All rights reserved.
#define BOOST_TEST_MODULE AllTrueTest
#include <boost/test/unit_test.hpp>
#include <poplar/Engine.hpp>
#include <poplibs_support/TestDevice.hpp>
#include <popops/AllTrue.hpp>
#include <popops/codelets.hpp>
#include <poputil/exceptions.hpp>

using namespace poplar;
using namespace poplar::program;
using namespace poputil;
using namespace popops;
using namespace poplibs_support;

#define DIM_SIZE 4

bool allTrueTest(bool in[DIM_SIZE]) {
  auto device = createTestDevice(TEST_TARGET);
  Graph graph(device.getTarget());
  popops::addCodelets(graph);

  Tensor tIn = graph.addVariable(BOOL, {DIM_SIZE}, "t1");
  graph.setTileMapping(tIn, 0);

  auto seq = Sequence();
  const auto tOut = allTrue(graph, tIn, seq, "");

  graph.createHostWrite("in", tIn);
  graph.createHostRead("out", tOut);

  bool out;
  Engine eng(graph, seq);
  device.bind([&](const Device &d) {
    eng.load(d);
    eng.writeTensor("in", in, &in[DIM_SIZE]);
    eng.run();
    eng.readTensor("out", &out, &out + 1);
  });
  return out;
}

BOOST_AUTO_TEST_CASE(AllTrue) {
  bool in[DIM_SIZE] = {true, true, true, true};
  BOOST_CHECK_EQUAL(allTrueTest(in), true);
}

BOOST_AUTO_TEST_CASE(NotAllTrue) {
  bool in[DIM_SIZE] = {true, false, false, true};
  BOOST_CHECK_EQUAL(allTrueTest(in), false);
}