topi_ewise_test.cc 405 Bytes
Newer Older
1
#include <tvm/tvm.h>
2
#include <topi/elemwise.h>
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
#include <gtest/gtest.h>

namespace topi {
TEST(Tensor, Basic) {
  using namespace tvm;
  Var m("m"), n("n"), l("l");
  Tensor A = placeholder({m, l}, Float(32), "A");
  auto C = topi::exp(A);
}
}

int main(int argc, char ** argv) {
  testing::InitGoogleTest(&argc, argv);
  testing::FLAGS_gtest_death_test_style = "threadsafe";
  return RUN_ALL_TESTS();
}