tiny.choir.ir.interfaces.ShapedTypeInterface

1pub const ShapedTypeInterface = struct {2    pub const interface_name = "ir.interface.shaped_type";3    pub const id: InterfaceId = base.interfaceId(interface_name);45    pub const VTable = struct {6        getRank: *const fn (type_ptr: *const anyopaque, ctx: *const ContextOpaque) ?usize,7        getShape: *const fn (type_ptr: *const anyopaque, ctx: *const ContextOpaque) ?[]const u64,8        getElementType: *const fn (type_ptr: *const anyopaque, ctx: *const ContextOpaque) ?IrType,9        getAddressSpaceTag: *const fn (type_ptr: *const anyopaque, ctx: *const ContextOpaque) ?u8,10    };1112    pub fn entry(vtable: *const VTable) InterfaceEntry {13        return .{ .id = id, .vtable = vtable };14    }1516    pub fn vtableFor(17        comptime getRank: *const fn (type_ptr: *const anyopaque, ctx: *const ContextOpaque) ?usize,18        comptime getShape: *const fn (type_ptr: *const anyopaque, ctx: *const ContextOpaque) ?[]const u64,19        comptime getElementType: *const fn (type_ptr: *const anyopaque, ctx: *const ContextOpaque) ?IrType,20        comptime getAddressSpaceTag: *const fn (type_ptr: *const anyopaque, ctx: *const ContextOpaque) ?u8,21    ) *const VTable {22        return &.{23            .getRank = getRank,24            .getShape = getShape,25            .getElementType = getElementType,26            .getAddressSpaceTag = getAddressSpaceTag,27        };28    }2930    pub fn entryFor(31        comptime getRank: *const fn (type_ptr: *const anyopaque, ctx: *const ContextOpaque) ?usize,32        comptime getShape: *const fn (type_ptr: *const anyopaque, ctx: *const ContextOpaque) ?[]const u64,33        comptime getElementType: *const fn (type_ptr: *const anyopaque, ctx: *const ContextOpaque) ?IrType,34        comptime getAddressSpaceTag: *const fn (type_ptr: *const anyopaque, ctx: *const ContextOpaque) ?u8,35    ) InterfaceEntry {36        return entry(vtableFor(getRank, getShape, getElementType, getAddressSpaceTag));37    }38};

1pub const ShapedTypeInterface = types.ShapedTypeInterface;