Function main

examples/transpose/run.cpp:210–241

}

int main() {
  char* version_str = getenv("TEST_VERSION");
  int version = version_str == NULL ? 2 : atoi(version_str);
    // 0 == cpu
    // 1 == naive transpose
    // 2 == tiling with shared memory

  size_t M, N;  // Matrix dimensions
  static constexpr int kTestSize = 2;
  if constexpr (kTestSize == 0) {
    // Tiny test
    M = 16;
    N = 32;
  } else if constexpr (kTestSize == 1) {
    // Small test
    M = 256;
    N = 512;
  } else {
    // Large test
    M = 4096;
    N = 2 * 4096;
  }

  std::unique_ptr<float[]> inputPtr = std::make_unique<float[]>(M * N);
  std::unique_ptr<float[]> outputPtr = std::make_unique<float[]>(N * M);

  initData(M, N, inputPtr);
  runTest(version, M, N, inputPtr, outputPtr);

  LOG(kDefLog, kInfo, "Done.");
  return 0;
}