[Arcilator] Timing scopes for JIT

Author: SimonEbner

tools/arcilator/arcilator.cpp

@@ -422,6 +422,7 @@ static LogicalResult processBuffer(
 #ifdef ARCILATOR_ENABLE_JIT
   // Handle JIT execution.
   if (outputFormat == OutputRunJIT) {
+    auto tsJit = ts.nest("JIT");
     if (runUntilBefore != UntilEnd || runUntilAfter != UntilEnd) {
       llvm::errs() << "full pipeline must be run for JIT execution\n";
       return failure();
@@ -460,6 +461,7 @@ static LogicalResult processBuffer(
     engineOptions.transformer = transformer;
     engineOptions.sharedLibPaths = sharedLibraries;
 
+    auto tsCompile = tsJit.nest("Compile");
     auto executionEngine =
         mlir::ExecutionEngine::create(module.get(), engineOptions);
     if (!executionEngine) {
@@ -480,7 +482,9 @@ static LogicalResult processBuffer(
           });
       return failure();
     }
+    tsCompile.stop();
 
+    auto tsExecute = tsJit.nest("Execute");
     void (*simulationFunc)(void **) = *expectedFunc;
     (*simulationFunc)(nullptr);