tiny.profiling.driver.measure

1const std = @import("std");2const catalog = @import("../root.zig").catalog;3const execute = @import("../root.zig").execute;4const host = @import("../root.zig").host;5const record = @import("../root.zig").record;6const MeasuredWorkload = @import("root.zig").model.MeasuredWorkload;7const PreparedWorkload = @import("root.zig").model.PreparedWorkload;8const WorkloadContext = @import("root.zig").model.WorkloadContext;9const absoluteArtifactEnv = @import("root.zig").paths.absoluteArtifactEnv;10const absoluteExecutionArtifactEnv = @import("root.zig").paths.absoluteExecutionArtifactEnv;11const aggregateMeasuredCommand = @import("root.zig").acquire.aggregateMeasuredCommand;12const runCaptureControlPlan = @import("root.zig").control.runCaptureControlPlan;13const runMeasuredCommand = @import("root.zig").acquire.runMeasuredCommand;14const writeCounterProgress = @import("root.zig").progress.writeCounterProgress;15const writeMeasureProgress = @import("root.zig").progress.writeMeasureProgress;1617pub fn runMeasuredWorkload(18    allocator: std.mem.Allocator,19    context: *const WorkloadContext,20    workload: catalog.Workload,21    prepared: PreparedWorkload,22    resets: *std.ArrayList(record.Reset),23) !MeasuredWorkload {24    const counter_options = context.request.host_lanes.counters;25    const counter_lane = if (prepared.wrapped.lane) |lane| switch (lane) {26        .counters => |value| value,27        else => null,28    } else null;29    if (context.request.capture_control.enabled()) {30        const capture_env = try absoluteArtifactEnv(allocator, prepared.paths);31        return try runCaptureControlPlan(32            allocator,33            context,34            workload,35            prepared,36            capture_env,37            resets,38        );39    }40    if (!context.tool_probe.available or counter_options == null or counter_lane == null) {41        var aggregate: ?execute.Result = null;42        var executions: std.ArrayList(record.MeasuredExecution) = .empty;43        var host_state_windows: std.ArrayList(host.state.Window) = .empty;44        var reset_failed = false;45        for (0..context.request.measure_repeat) |repeat_index| {46            if (context.progress_output == .terminal) {47                try writeMeasureProgress(48                    allocator,49                    workload.name,50                    repeat_index,51                    context.request.measure_repeat,52                );53            }54            const execution_paths = try record.executionPaths(55                allocator,56                prepared.paths,57                context.request.measure_repeat,58                repeat_index + 1,59            );60            const artifacts = try absoluteExecutionArtifactEnv(61                allocator,62                execution_paths,63            );64            const measurement = try runMeasuredCommand(65                allocator,66                context,67                workload,68                prepared,69                prepared.wrapped.argv,70                artifacts,71                .measurement,72                .measurement,73                repeat_index + 1,74                resets,75            );76            const result = measurement.execution;77            if (measurement.measured) {78                try executions.append(allocator, .{79                    .index = repeat_index + 1,80                    .result = result,81                    .paths = execution_paths,82                });83            } else {84                reset_failed = true;85            }86            if (measurement.host_state) |window| {87                try host_state_windows.append(allocator, window);88            }89            aggregate = aggregateMeasuredCommand(aggregate, measurement);90            if (result.exit_code != 0) break;91        }92        return .{93            .aggregate = aggregate.?,94            .executions = try executions.toOwnedSlice(allocator),95            .host_state_windows = try host_state_windows.toOwnedSlice(allocator),96            .reset_failed = reset_failed,97        };98    }99    const options = counter_options.?;100    const lane = counter_lane.?;101    std.debug.assert(lane.runs.len == lane.event_groups.len * options.repeat);102    var aggregate: ?execute.Result = null;103    var executions: std.ArrayList(record.MeasuredExecution) = .empty;104    var host_state_windows: std.ArrayList(host.state.Window) = .empty;105    var reset_failed = false;106    for (lane.runs, 0..) |run, run_index| {107        if (context.progress_output == .terminal) {108            try writeCounterProgress(109                allocator,110                workload.name,111                lane,112                options,113                run,114                run_index,115            );116        }117        const argv = if (run_index == 0)118            prepared.wrapped.argv119        else120            try host.counters.wrapRunArgv(121                allocator,122                options,123                prepared.execution_plan.argv,124                lane,125                run_index,126            );127        const execution_paths = try record.executionPaths(128            allocator,129            prepared.paths,130            lane.runs.len,131            run_index + 1,132        );133        const artifacts = try absoluteExecutionArtifactEnv(134            allocator,135            execution_paths,136        );137        const measurement = try runMeasuredCommand(138            allocator,139            context,140            workload,141            prepared,142            argv,143            artifacts,144            .measurement,145            .counter,146            run_index + 1,147            resets,148        );149        const result = measurement.execution;150        if (measurement.measured) {151            try executions.append(allocator, .{152                .index = run_index + 1,153                .result = result,154                .paths = execution_paths,155            });156        } else {157            reset_failed = true;158        }159        if (measurement.host_state) |window| {160            try host_state_windows.append(allocator, window);161        }162        aggregate = aggregateMeasuredCommand(aggregate, measurement);163        if (result.exit_code != 0) break;164    }165    return .{166        .aggregate = aggregate.?,167        .executions = try executions.toOwnedSlice(allocator),168        .host_state_windows = try host_state_windows.toOwnedSlice(allocator),169        .reset_failed = reset_failed,170    };171}

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