Function main

examples/matmul/run.cpp:905–957

}

int main() {
  char* version_str = getenv("MATMUL_VERSION");
  int version = version_str == NULL ? 10 : atoi(version_str);
    // 1 == f32: No-Op
    // 2 == f32: naive matmul
    // 3 == f32: tiling
    // 4 == f32: 1D blocktiling
    // 5 == f32: 2D blocktiling
    // 6 == f32: 1D blocktiling with loop unrolling
    // 7 == f32: 2D blocktiling with loop unrolling
    // 8 == f32: 2D blocktiling with loop unrolling and vectorization
    // 9 == f32: 2D blocktiling with loop unrolling, vectorization and transpose
    // 10 == f16: 2D blocktiling with loop unrolling and vectorization (default)
    // 11 == f16: 2D blocktiling with loop unrolling, vectorization and transpose
  bool enableF16 = version == 10 || version ==11;
  bool transposedInput = version == 9 || version == 11;
  NumType numtype = enableF16 ? kf16 : kf32;

  size_t M, K, N;  // Matrix dimensions
  char* kTestSize_str = getenv("MATMUL_SIZE");
  int kTestSize = kTestSize_str == NULL ? 2 : atoi(kTestSize_str);
  if (kTestSize == 0) {
    // Tiny test
    M = 32;
    K = 32;
    N = 32;
  } else if (kTestSize == 1) {
    // Small test
    M = 256;
    K = 128;
    N = 512;
  } else {
    // Large test
    M = 4096;
    K = 4096;
    N = 2 * 4096;
  }

#ifdef METAL_PROFILER
  startCapture();
#endif
  if (enableF16) {
    runTestWithCheck<half>(version, M, K, N, transposedInput, kTestSize, numtype);
  } else {
    runTestWithCheck<float>(version, M, K, N, transposedInput, kTestSize, numtype);
  }
#ifdef METAL_PROFILER
  stopCapture();
#endif

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