tiny.choir.backends.gpu.spirv.emitter.plan

1const std = @import("std");2const abi = @import("choir_abi");3const choir = @import("../../../../root.zig");45const codegen = @import("codegen.zig");67const ir = choir.ir;89/// SPIR-V words for one module, and the push-constant layout the emitter gave10/// the requested entry.11pub const Emission = struct {12    words: []u32,13    push_constants: abi.PushConstants,14};1516/// Emits `module` and returns the layout of `entry_name`'s push-constant17/// block. A hand-written `spirv.module` declares no block, so its layout is18/// empty. `controls` contains only float modes the selected device reports.19pub fn emitPlanWords(20    result_allocator: std.mem.Allocator,21    module: *ir.Operation,22    entry_name: []const u8,23    limits: abi.Limits,24    controls: codegen.FloatControls,25) abi.Error!Emission {26    std.debug.assert(entry_name.len > 0);27    std.debug.assert(limits.valid());28    var emitter = codegen.SpirvCodegen.init(result_allocator);29    defer emitter.deinit();30    emitter.limits = limits;31    emitter.entry_name = entry_name;32    emitter.float_controls = controls;33    const words = emitter.emitModuleWords(module) catch |err| return mapCodegenError(err);34    return .{ .words = words, .push_constants = emitter.entry_push_constants orelse .{} };35}3637fn mapCodegenError(err: codegen.SpirvCodegenError) abi.Error {38    return switch (err) {39        error.OutOfMemory => error.OutOfMemory,40        error.UnsupportedOperation, error.UnsupportedStageMemory, error.UnsupportedHelperSignature => error.UnsupportedOperation,41        error.UnsupportedFunctionSignature,42        error.UnsupportedType,43        error.UnsupportedMask,44        error.UnsupportedAddressSpace,45        error.UnsupportedControlFlow,46        => error.CapabilityMismatch,47        else => error.InvalidArtifact,48    };49}5051test {52    std.testing.refAllDecls(@This());53}

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