tiny.accy.choir.gpu

1const std = @import("std");2const choir = @import("choir");3const contract = @import("contract.zig");4const dispatch = @import("dispatch.zig");5const memory = @import("memory.zig");6const semantic = @import("semantic.zig");7const tensor = @import("tensor.zig");89const ir = choir.ir;10const passes = choir.passes;1112pub const product_name = "accy.kernel/gpu";1314pub const KernelModule = @import("root.zig").publication.Module(.kernel);1516pub const KernelJob = struct {17    allocator: std.mem.Allocator,18    memory_module: *memory.MemoryJob,19    choir_module: *ir.Operation,20    observed_fingerprint: u64,2122    pub fn init(23        allocator: std.mem.Allocator,24        memory_module: *memory.MemoryJob,25    ) !*KernelJob {26        const module = try allocator.create(KernelJob);27        errdefer allocator.destroy(module);28        module.* = .{29            .allocator = allocator,30            .memory_module = memory_module,31            .choir_module = memory_module.choir_module,32            .observed_fingerprint = 0,33        };34        try module.verify();35        module.observed_fingerprint = try choir.operationFingerprint(allocator, module.choir_module);36        return module;37    }3839    pub fn context(self: *KernelJob) *ir.Context {40        return self.memory_module.context();41    }4243    pub fn deinit(self: *KernelJob) void {44        self.memory_module.deinit();45        const allocator = self.allocator;46        allocator.destroy(self);47    }4849    pub fn verify(self: *KernelJob) !void {50        try self.memory_module.verify();51    }5253    pub fn fingerprint(self: *const KernelJob) u64 {54        return self.observed_fingerprint;55    }5657    pub fn passContext(self: *KernelJob) passes.PassContext {58        return self.memory_module.passContext();59    }60};6162test "kernel module owns memory product" {63    const allocator = std.testing.allocator;6465    var builder = try semantic.Builder.init(allocator, semantic.Builder.ContextLimits.standard);66    defer builder.deinit();6768    const ty = try builder.tensor(.f32, &.{4});69    var function = try builder.beginFunction("kernel_add", &.{ ty, ty }, &.{ty});70    const sum = try function.add(function.parameter(0), function.parameter(1));71    try function.return_(&.{sum});72    try function.finish();7374    const semantic_module = try builder.finish();75    var contract_module = try contract.ContractJob.init(allocator, semantic_module);76    var contract_owned = true;77    errdefer if (contract_owned) contract_module.deinit();7879    var analysis_cache = passes.AnalysisCache.init(allocator, null);80    var cache_owned = true;81    errdefer if (cache_owned) analysis_cache.deinit();8283    var tensor_module = try tensor.TensorJob.init(allocator, contract_module, analysis_cache);84    contract_owned = false;85    cache_owned = false;86    var tensor_owned = true;87    errdefer if (tensor_owned) tensor_module.deinit();8889    var dispatch_module = try dispatch.DispatchJob.init(allocator, tensor_module);90    tensor_owned = false;91    var dispatch_owned = true;92    errdefer if (dispatch_owned) dispatch_module.deinit();9394    var memory_module = try memory.MemoryJob.init(allocator, dispatch_module);95    dispatch_owned = false;96    var memory_owned = true;97    errdefer if (memory_owned) memory_module.deinit();9899    var kernel_module = try KernelJob.init(allocator, memory_module);100    memory_owned = false;101    defer kernel_module.deinit();102103    try std.testing.expectEqualStrings(product_name, "accy.kernel/gpu");104    try kernel_module.verify();105    try std.testing.expectEqual(try choir.operationFingerprint(allocator, kernel_module.choir_module), kernel_module.fingerprint());106}

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