diff --git a/enzyme/benchmarks/ReverseMode/adbench/gmm.h b/enzyme/benchmarks/ReverseMode/adbench/gmm.h index 24da48fc791..1c6c203a964 100644 --- 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 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;