File tree Expand file tree Collapse file tree 2 files changed +4
-1
lines changed
android/benchmark/app/src/main/java/org/pytorch/minibench Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -42,7 +42,7 @@ protected void onCreate(Bundle savedInstanceState) {
4242 .findFirst ()
4343 .get ();
4444
45- int numIter = intent .getIntExtra ("num_iter" , 10 );
45+ int numIter = intent .getIntExtra ("num_iter" , 20 );
4646
4747 // TODO: Format the string with a parsable format
4848 Stats stats = new Stats ();
Original file line number Diff line number Diff line change @@ -85,7 +85,10 @@ @implementation GenericTests
8585 XCTFail (" Unsupported tag %i at input %d" , *input_tag, index);
8686 }
8787 }
88+ XCTMeasureOptions *options = [[XCTMeasureOptions alloc ] init ];
89+ options.iterationCount = 20 ; // Set the number of iterations to 20
8890 [testCase measureWithMetrics: @[ [XCTClockMetric new ], [XCTMemoryMetric new ] ]
91+ options: options
8992 block: ^{
9093 XCTAssertEqual (module ->forward ().error (), Error::Ok);
9194 }];
You can’t perform that action at this time.
0 commit comments