Skip to content

Commit dc88a6b

Browse files
committed
Minibench fix
1 parent 33844d3 commit dc88a6b

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/BenchmarkActivity.java

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -111,13 +111,13 @@ public BenchmarkHandler(Looper looper, BenchmarkActivity benchmarkActivity) {
111111
@Override
112112
public void handleMessage(android.os.Message msg) {
113113
if (msg.what == MESSAGE_RUN_BENCHMARK) {
114-
mModelRunner.runBenchmark(
115-
mBenchmarkActivity.mModel,
116-
mBenchmarkActivity.mNumWarmupIter,
117-
mBenchmarkActivity.mNumIter,
118-
mBenchmarkActivity.mResult);
119-
120114
if (mBenchmarkActivity.mTokenizerPath == null) {
115+
mModelRunner.runBenchmark(
116+
mBenchmarkActivity.mModel,
117+
mBenchmarkActivity.mNumWarmupIter,
118+
mBenchmarkActivity.mNumIter,
119+
mBenchmarkActivity.mResult);
120+
121121
mBenchmarkActivity.writeResult();
122122
} else {
123123
this.sendEmptyMessage(MESSAGE_LLM_RUN_BENCHMARK);

0 commit comments

Comments
 (0)