bench gmm: use env BENCH_LARGE=1 to run the larger (slow) tests

EnzymeAD · Oct 29, 2024 · a2d9fed · a2d9fed
1 parent 8e6dabc
commit a2d9fed
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/enzyme/benchmarks/ReverseMode/adbench/gmm.h b/enzyme/benchmarks/ReverseMode/adbench/gmm.h
@@ -168,8 +168,10 @@ int main(const int argc, const char* argv[]) {
     std::vector<std::string> paths;// = { "1k/gmm_d10_K100.txt" };
 
     getTests(paths, "data/1k", "1k/");
-    //getTests(paths, "data/2.5k", "2.5k/");
-    //getTests(paths, "data/10k", "10k/");
+    if (std::getenv("BENCH_LARGE")) {
+      getTests(paths, "data/2.5k", "2.5k/");
+      getTests(paths, "data/10k", "10k/");
+    }
 
     std::ofstream jsonfile("results.json", std::ofstream::trunc);
     json test_results;