From 94960f123e85c6aac59e66ffb4c030d71de4c202 Mon Sep 17 00:00:00 2001
From: Joe Vilches <joevilches@meta.com>
Date: Mon, 5 Feb 2024 11:48:07 -0800
Subject: [PATCH] main() function to run a benchmark (#1564)

Summary:
Pull Request resolved: https://github.com/facebook/yoga/pull/1564

tsia

Reviewed By: NickGerleman

Differential Revision: D53028962

fbshipit-source-id: dee2670eaa4fe0ab8b6e2b9e1bbe4356bb2c0735
---
 benchmark/Benchmark.cpp | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/benchmark/Benchmark.cpp b/benchmark/Benchmark.cpp
index 054952a8..b517b83e 100644
--- a/benchmark/Benchmark.cpp
+++ b/benchmark/Benchmark.cpp
@@ -391,6 +391,14 @@ void benchmark(std::filesystem::path& capturesDir) {
 
 } // namespace facebook::yoga
 
-int main() {
+int main(int argc, char* argv[]) {
+  if (argc == 2) {
+    std::filesystem::path capturesDir = argv[argc - 1];
+    facebook::yoga::benchmark(capturesDir);
+  } else {
+    throw std::invalid_argument("Expecting a path as an argument");
+    return 1;
+  }
+
   return 0;
 }