Skip to content

Commit f46de48

Browse files
committed
[Arcilator] Timing scopes for JIT
1 parent 4d81b97 commit f46de48

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tools/arcilator/arcilator.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -420,6 +420,8 @@ static LogicalResult processBuffer(
420420
return failure();
421421

422422
#ifdef ARCILATOR_ENABLE_JIT
423+
auto tsJit = ts.nest("JIT");
424+
423425
// Handle JIT execution.
424426
if (outputFormat == OutputRunJIT) {
425427
if (runUntilBefore != UntilEnd || runUntilAfter != UntilEnd) {
@@ -460,6 +462,7 @@ static LogicalResult processBuffer(
460462
engineOptions.transformer = transformer;
461463
engineOptions.sharedLibPaths = sharedLibraries;
462464

465+
auto tsCompile = tsJit.nest("Compile");
463466
auto executionEngine =
464467
mlir::ExecutionEngine::create(module.get(), engineOptions);
465468
if (!executionEngine) {
@@ -480,7 +483,9 @@ static LogicalResult processBuffer(
480483
});
481484
return failure();
482485
}
486+
tsCompile.stop();
483487

488+
auto tsExecute = tsJit.nest("Execute");
484489
void (*simulationFunc)(void **) = *expectedFunc;
485490
(*simulationFunc)(nullptr);
486491

0 commit comments

Comments
 (0)