From 56db8eee7c2b078a1767d97dcd7c5633e9c05620 Mon Sep 17 00:00:00 2001 From: Guang Yang Date: Mon, 14 Oct 2024 17:33:08 -0700 Subject: [PATCH] Increase number of iterations for benchmarking --- .../src/main/java/org/pytorch/minibench/BenchmarkActivity.java | 2 +- extension/benchmark/apple/Benchmark/Tests/GenericTests.mm | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/BenchmarkActivity.java b/extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/BenchmarkActivity.java index 2397bcfb851..e2d46f8e8d0 100644 --- a/extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/BenchmarkActivity.java +++ b/extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/BenchmarkActivity.java @@ -42,7 +42,7 @@ protected void onCreate(Bundle savedInstanceState) { .findFirst() .get(); - int numIter = intent.getIntExtra("num_iter", 10); + int numIter = intent.getIntExtra("num_iter", 50); // TODO: Format the string with a parsable format Stats stats = new Stats(); diff --git a/extension/benchmark/apple/Benchmark/Tests/GenericTests.mm b/extension/benchmark/apple/Benchmark/Tests/GenericTests.mm index ce685335767..f6c6927e78e 100644 --- a/extension/benchmark/apple/Benchmark/Tests/GenericTests.mm +++ b/extension/benchmark/apple/Benchmark/Tests/GenericTests.mm @@ -85,7 +85,10 @@ @implementation GenericTests XCTFail("Unsupported tag %i at input %d", *input_tag, index); } } + XCTMeasureOptions *options = [[XCTMeasureOptions alloc] init]; + options.iterationCount = 20; [testCase measureWithMetrics:@[ [XCTClockMetric new], [XCTMemoryMetric new] ] + options:options block:^{ XCTAssertEqual(module->forward().error(), Error::Ok); }];