tiny.accy.kernel.library.random.base

1const choir_abi = @import("choir_abi");2pub const std = @import("std");34pub const artifact_product = @import("../../../artifact/model/root.zig");5pub const shape = @import("../../../choir/shape/root.zig");6pub const library = @import("../root.zig");7pub const kernel = @import("../../root.zig");89pub const entry = library.entry;10pub const extent_mod = library.extent;11pub const geometry_mod = library.geometry;12pub const tuning = library.tuning;1314pub const DType = choir_abi.DType;15pub const indexExtent = extent_mod.indexExtent;16pub const runtimeExtentArgument = extent_mod.runtimeExtentArgument;17pub const random_thread_caps = geometry_mod.ThreadCaps1D{};1819pub fn randomRuntimeExtentBounds() shape.Bounds {20    return .{ .min = 1, .max = extent_mod.runtime_extent_max };21}2223pub fn randomDerivedLaunch(threads: u32, lanes: u64) !artifact_product.KernelCallLaunch {24    if (threads == 0) return error.KernelLibraryLaunchThreadgroupMustBeNonzero;25    const divisor = std.math.cast(u32, @as(u64, threads) * lanes) orelse26        return error.KernelLibraryLaunchThreadgroupMustBeNonzero;27    return .{ .derived = .{28        .grid = .{29            .{ .runtime_u32_ceil_div = .{ .argument_index = 0, .divisor = divisor } },30            .{ .fixed = 1 },31            .{ .fixed = 1 },32        },33        .threadgroup = .{ threads, 1, 1 },34    } };35}3637pub fn randomFoldDerivedLaunch(threads: u32) !artifact_product.KernelCallLaunch {38    if (threads == 0) return error.KernelLibraryLaunchThreadgroupMustBeNonzero;39    return .{ .derived = .{40        .grid = .{41            .{ .runtime_u32_ceil_div = .{ .argument_index = 0, .divisor = threads } },42            .{ .fixed = 1 },43            .{ .fixed = 1 },44        },45        .threadgroup = .{ threads, 1, 1 },46    } };47}4849pub fn randomShapeFamily(backing_allocator: std.mem.Allocator, name: []const u8, count_axis: []const u8) !shape.Family {50    var builder = try shape.Builder.init(backing_allocator, name);51    errdefer builder.deinit();5253    const count = try builder.symbol(count_axis);54    const count_expr = try builder.symbolExpression(count);55    _ = try builder.tensor("out", &.{count_expr});56    try builder.assumeBounds(count_expr, randomRuntimeExtentBounds());5758    return builder.finish();59}6061pub fn ceilDivComptime(comptime numerator: u64, comptime denominator: u32) u32 {62    return @intCast(numerator / denominator + @as(u64, @intFromBool(numerator % denominator != 0)));63}

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