Function main

experimental/legacy/transformer/test_kernels.cpp:271–283

}

int main(int argc, char **argv) {
  Context ctx = createContext();

  testTensorPool(ctx);
  testResidual(ctx);
  testHadamard(ctx);
  testMatmul(ctx);
  testGelu(ctx);
  testLayerNorm(ctx);
  testSoftmax(ctx);

  LOG(kDefLog, kInfo, "Done with all tests");
}