tiny.accy.kernel.interpret.surface

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

1const std = @import("std");23const policy = @import("policy.zig");4const program = @import("../program/root.zig");5const spec = @import("spec.zig");67pub const ExclusionKind = enum {8    lifecycle,9    diagnostic_finish,10    schedule_construction,11};1213pub const Exclusion = struct {14    name: []const u8,15    kind: ExclusionKind,16};1718pub const excluded_methods = [_]Exclusion{19    .{ .name = "init", .kind = .lifecycle },20    .{ .name = "deinit", .kind = .lifecycle },21    .{ .name = "finishWithDiagnostic", .kind = .diagnostic_finish },22    .{ .name = "axis", .kind = .schedule_construction },23    .{ .name = "split", .kind = .schedule_construction },24    .{ .name = "bind", .kind = .schedule_construction },25    .{ .name = "vectorize", .kind = .schedule_construction },26    .{ .name = "unroll", .kind = .schedule_construction },27};2829pub fn exclusion(comptime name: []const u8) ?Exclusion {30    inline for (excluded_methods) |item| {31        if (std.mem.eql(u8, item.name, name)) return item;32    }33    return null;34}3536pub fn supports(comptime name: []const u8) bool {37    return isFnDecl(program.Builder, name) and exclusion(name) == null and @hasDecl(DefaultLayer(), name);38}3940pub fn assertLayerCoversBuilder(comptime LayerType: type) void {41    assertExclusionsReferToBuilderMethods();42    inline for (@typeInfo(program.Builder).@"struct".decl_names) |decl_name| {43        if (!isFnDecl(program.Builder, decl_name)) continue;44        if (exclusion(decl_name) != null) continue;45        if (!@hasDecl(LayerType, decl_name)) {46            @compileError("kernel.interpret.Layer missing typed builder method: " ++ decl_name);47        }48    }49}5051fn DefaultLayer() type {52    return spec.Layer(*program.Builder, struct {}, policy.lower);53}5455fn assertExclusionsReferToBuilderMethods() void {56    inline for (excluded_methods) |item| {57        if (!isFnDecl(program.Builder, item.name)) {58            @compileError("kernel.interpret surface exclusion is not a typed builder method: " ++ item.name);59        }60    }61}6263fn isFnDecl(comptime Type: type, comptime name: []const u8) bool {64    if (!@hasDecl(Type, name)) return false;65    return switch (@typeInfo(@TypeOf(@field(Type, name)))) {66        .@"fn" => true,67        else => false,68    };69}7071test "kernel interpret Layer covers typed builder body surface" {72    assertLayerCoversBuilder(DefaultLayer());73    try std.testing.expect(supports("forScope"));74    try std.testing.expect(supports("tan"));75    try std.testing.expect(exclusion("axis").?.kind == .schedule_construction);76}