tiny.profiling.driver.publish

1const std = @import("std");2const catalog = @import("../root.zig").catalog;3const host = @import("../root.zig").host;4const ingest = @import("../root.zig").ingest;5const order = @import("../root.zig").order;6const record = @import("../root.zig").record;7const reducer = @import("../root.zig").reduction;8const CompletedWorkload = @import("root.zig").model.CompletedWorkload;9const PreparedWorkload = @import("root.zig").model.PreparedWorkload;10const WorkloadContext = @import("root.zig").model.WorkloadContext;11const benchmarkReductionDomain = @import("root.zig").artifact.benchmarkReductionDomain;12const collectCaptures = @import("root.zig").artifact.collectCaptures;13const collectMeasuredArtifacts = @import("root.zig").artifact.collectMeasuredArtifacts;14const hasControlResult = @import("root.zig").artifact.hasControlResult;15const perturbationStructuredPaths = @import("root.zig").artifact.perturbationStructuredPaths;16const reduceMeasuredBenchmarks = @import("root.zig").artifact.reduceMeasuredBenchmarks;17const retainedArtifactPaths = @import("root.zig").artifact.retainedArtifactPaths;18const writeCaptureCaveats = @import("root.zig").progress.writeCaptureCaveats;1920pub fn recordCompletedWorkload(21    allocator: std.mem.Allocator,22    context: *const WorkloadContext,23    workload: catalog.Workload,24    index: usize,25    prepared: PreparedWorkload,26    completed_value: CompletedWorkload,27    acquisition: order.Context,28    started_unix_ns: i128,29    finished_unix_ns: i128,30) !i64 {31    var completed = completed_value;32    const scope_label = if (prepared.execution_plan.direct) host.direct_scope else host.scope;33    const structured = try collectMeasuredArtifacts(34        allocator,35        context,36        workload,37        prepared,38        completed.executions,39    );40    const benchmark_process_domain = benchmarkReductionDomain(41        context.request,42        workload.surface,43        workload.tracesAllocationsByDefault(),44        prepared.execution_plan,45    );46    const benchmark_process_reduction = try reduceMeasuredBenchmarks(47        allocator,48        benchmark_process_domain,49        completed.executions,50    );51    const captures = try collectCaptures(52        allocator,53        context,54        workload,55        prepared,56        completed,57        scope_label,58        structured.max_benchmark_allocated_bytes,59    );60    try writeCaptureCaveats(allocator, workload.name, captures);61    if (completed.perturbation) |*measurement| {62        if (hasControlResult(measurement.pairs)) {63            const control_structured = try ingest.collect(64                allocator,65                workload,66                perturbationStructuredPaths(measurement.control_artifacts),67            );68            measurement.control_structured_rows = control_structured.rows;69            measurement.control_structured_parse_errors = control_structured.parse_errors;70        }71    }72    try appendWorkloadRecord(73        allocator,74        context,75        workload,76        index,77        prepared,78        completed,79        acquisition,80        scope_label,81        captures,82        structured,83        benchmark_process_domain,84        benchmark_process_reduction,85        started_unix_ns,86        finished_unix_ns,87    );88    return completed.execution.exit_code;89}9091fn appendWorkloadRecord(92    allocator: std.mem.Allocator,93    context: *const WorkloadContext,94    workload: catalog.Workload,95    index: usize,96    prepared: PreparedWorkload,97    completed: CompletedWorkload,98    acquisition: order.Context,99    scope_label: []const u8,100    captures: []const host.Capture,101    structured: ingest.Summary,102    benchmark_process_domain: reducer.DomainDeclaration,103    benchmark_process_reduction: reducer.Result,104    started_unix_ns: i128,105    finished_unix_ns: i128,106) !void {107    std.debug.assert(index < context.request.selection.count());108    std.debug.assert(prepared.command.len > 0);109    std.debug.assert(acquisition.valid());110    const artifacts = retainedArtifactPaths(prepared.paths, completed.executions);111    try record.appendWorkload(allocator, context.paths, context.metadata, .{112        .index = index,113        .acquisition = acquisition,114        .workload = workload,115        .runtime = prepared.runtime,116        .argv = prepared.wrapped.argv,117        .command = prepared.command,118        .cwd = prepared.execution_plan.cwd,119        .scope = scope_label,120        .causal = context.request.control.causal,121        .tracy = context.request.tracy,122        .trace_allocations = context.request.trace_allocations or123            workload.tracesAllocationsByDefault(),124        .bench_min_time_ns = context.request.control.min_time_ns,125        .causal_filter = context.request.control.filter,126        .setup = completed.setup,127        .prepare = completed.prepare,128        .artifact_root = prepared.paths.root,129        .command_path = prepared.paths.command,130        .result_path = prepared.paths.result,131        .stdout_path = artifacts.stdout,132        .stderr_path = artifacts.stderr,133        .bench_jsonl_path = artifacts.bench_jsonl,134        .coz_jsonl_path = artifacts.coz_jsonl,135        .coz_analysis_path = artifacts.coz_analysis,136        .tracy_jsonl_path = artifacts.tracy_jsonl,137        .tracy_summary_path = artifacts.tracy_summary,138        .allocations_path = artifacts.allocations,139        .allocation_repetitions_path = prepared.paths.allocation_repetitions,140        .structured_path = prepared.paths.structured,141        .structured_rows = structured.rows,142        .structured_parse_errors = structured.parse_errors,143        .started_unix_ns = started_unix_ns,144        .finished_unix_ns = finished_unix_ns,145        .execution = completed.execution,146        .executions = completed.executions,147        .benchmark_process_domain = benchmark_process_domain,148        .benchmark_process_reduction = benchmark_process_reduction,149        .host_state_windows = completed.host_state_windows,150        .warmup_executions = completed.warmup_executions,151        .warmup_paths = prepared.paths.warmup,152        .resets = completed.resets,153        .perturbation = completed.perturbation,154        .captures = captures,155    });156}

1pub const publish = @import("publish.zig");