tiny.gpu.KernelArtifact

1pub const KernelArtifact = struct {2    allocator: Allocator,3    backend: BackendKind,4    format: ArtifactFormat,5    entry_name: []const u8,6    argument_count: u32,7    scalar_argument_count: u32 = 0,8    diagnostic_id: ?[]const u8 = null,9    /// Requirements and push-constant layout the compiler recorded for this10    /// entry. Load checks the requirements; launch packs through the layout.11    interface: choir_abi.Interface = .{},12    payload: ArtifactPayload = .none,13    payload_ownership: PayloadOwnership = .borrowed,1415    pub fn init(allocator: Allocator, desc: KernelArtifactDesc) Allocator.Error!KernelArtifact {16        const entry_name = try dupeString(allocator, desc.entry_name);17        errdefer freeString(allocator, entry_name);1819        return .{20            .allocator = allocator,21            .backend = desc.backend,22            .format = desc.format,23            .entry_name = entry_name,24            .argument_count = desc.argument_count,25            .scalar_argument_count = desc.scalar_argument_count,26            .diagnostic_id = try dupeOpt(allocator, desc.diagnostic_id),27            .interface = desc.interface,28        };29    }3031    pub fn bufferArgumentCount(self: *const KernelArtifact) BackendError!u32 {32        if (self.scalar_argument_count > self.argument_count) return error.InvalidArtifact;33        return self.argument_count - self.scalar_argument_count;34    }3536    pub fn deinit(self: *KernelArtifact) void {37        self.clearPayload();38        freeString(self.allocator, self.entry_name);39        freeOpt(self.allocator, self.diagnostic_id);40        self.* = undefined;41    }4243    pub fn setBorrowedBytes(self: *KernelArtifact, bytes: []const u8) void {44        self.clearPayload();45        self.payload = .{ .bytes = bytes };46        self.payload_ownership = .borrowed;47    }4849    pub fn setOwnedBytes(self: *KernelArtifact, bytes: []const u8) BackendError!void {50        const owned = self.allocator.dupe(u8, bytes) catch return error.OutOfMemory;51        self.clearPayload();52        self.payload = .{ .bytes = owned };53        self.payload_ownership = .owned;54    }5556    pub fn setBorrowedWords(self: *KernelArtifact, words: []const u32) void {57        self.clearPayload();58        self.payload = .{ .words_u32 = words };59        self.payload_ownership = .borrowed;60    }6162    pub fn setOwnedWords(self: *KernelArtifact, words: []const u32) BackendError!void {63        const owned = self.allocator.dupe(u32, words) catch return error.OutOfMemory;64        self.clearPayload();65        self.payload = .{ .words_u32 = owned };66        self.payload_ownership = .owned;67    }6869    pub fn setBorrowedText(self: *KernelArtifact, text: []const u8) void {70        self.clearPayload();71        self.payload = .{ .text = text };72        self.payload_ownership = .borrowed;73    }7475    pub fn setOwnedText(self: *KernelArtifact, text: []const u8) BackendError!void {76        const owned = self.allocator.dupe(u8, text) catch return error.OutOfMemory;77        self.clearPayload();78        self.payload = .{ .text = owned };79        self.payload_ownership = .owned;80    }8182    pub fn setExternalPayload(self: *KernelArtifact, payload: ExternalPayload, ownership: PayloadOwnership) BackendError!void {83        if (ownership == .owned and payload.deinit_fn == null) return error.MissingPayloadDeinit;84        self.clearPayload();85        self.payload = .{ .external = payload };86        self.payload_ownership = ownership;87    }8889    fn clearPayload(self: *KernelArtifact) void {90        if (self.payload_ownership == .owned) {91            switch (self.payload) {92                .bytes => |bytes| freeString(self.allocator, bytes),93                .words_u32 => |words| self.allocator.free(@constCast(words)),94                .text => |text| freeString(self.allocator, text),95                .external => |payload| if (payload.deinit_fn) |deinit_fn| deinit_fn(self.allocator, payload.ptr),96                .none => {},97            }98        }99        self.payload = .none;100        self.payload_ownership = .borrowed;101    }102};

1pub const KernelArtifact = contract.KernelArtifact;