Function main

src/test.cpp:312–325

}

int main(int argc, char* argv[]) {
  if (argc == 2 && std::string(argv[1]) == "-b") {
    std::cout << "Running memory benchmark" << std::endl;
    mem_bench();
  } else if (argc == 2 && std::string(argv[1]) == "-b2") {
    std::cout << "Running memory benchmark 2" << std::endl;
    mem_bench2();
  } else {
    test_attn();
    test_matmul();
  }
  std::cout << "All tests passed" << std::endl;
  return 0;
}