| ... | @@ -2,6 +2,8 @@ const std = @import("std"); | ... | @@ -2,6 +2,8 @@ const std = @import("std"); |
| 2 | const Allocator = std.mem.Allocator; | 2 | const Allocator = std.mem.Allocator; |
| 3 | const log = std.log.scoped(.codegen); | 3 | const log = std.log.scoped(.codegen); |
| 4 | | 4 | |
| | 5 | const Target = std.Target; |
| | 6 | |
| 5 | const spec = @import("spirv/spec.zig"); | 7 | const spec = @import("spirv/spec.zig"); |
| 6 | const Module = @import("../Module.zig"); | 8 | const Module = @import("../Module.zig"); |
| 7 | const Decl = Module.Decl; | 9 | const Decl = Module.Decl; |
| ... | @@ -63,6 +65,43 @@ pub const DeclGen = struct { | ... | @@ -63,6 +65,43 @@ pub const DeclGen = struct { |
| 63 | return error.AnalysisFail; | 65 | return error.AnalysisFail; |
| 64 | } | 66 | } |
| 65 | | 67 | |
| | 68 | /// SPIR-V requires enabling specific integer sizes through capabilities, and so if they are not enabled, we need |
| | 69 | /// to emulate them in other instructions/types. This function returns, given an integer bit width (signed or unsigned, sign |
| | 70 | /// included), the width of the underlying type which represents it, given the enabled features for the current target. |
| | 71 | /// If the result is `null`, the largest type the target platform supports natively is not able to perform computations using |
| | 72 | /// that size. In this case, multiple elements of the largest type should be used. |
| | 73 | /// The backing type will be chosen as the smallest supported integer larger or equal to it in number of bits. |
| | 74 | /// The result is valid to be used with OpTypeInt. |
| | 75 | /// TODO: The extension SPV_INTEL_arbitrary_precision_integers allows any integer size (at least up to 32 bits). |
| | 76 | /// TODO: This probably needs an ABI-version as well (especially in combination with SPV_INTEL_arbitrary_precision_integers). |
| | 77 | fn backingIntBits(self: *DeclGen, bits: u32) ?u32 { |
| | 78 | // TODO: Figure out what to do with u0/i0. |
| | 79 | std.debug.assert(bits != 0); |
| | 80 | |
| | 81 | const target = self.module.getTarget(); |
| | 82 | |
| | 83 | // 8, 16 and 64-bit integers require the Int8, Int16 and Inr64 capabilities respectively. |
| | 84 | const ints = [_]struct{ bits: u32, feature: ?Target.spirv.Feature } { |
| | 85 | .{ .bits = 8, .feature = .Int8 }, |
| | 86 | .{ .bits = 16, .feature = .Int16 }, |
| | 87 | .{ .bits = 32, .feature = null }, |
| | 88 | .{ .bits = 64, .feature = .Int64 }, |
| | 89 | }; |
| | 90 | |
| | 91 | for (ints) |int| { |
| | 92 | const has_feature = if (int.feature) |feature| |
| | 93 | Target.spirv.featureSetHas(target.cpu.features, feature) |
| | 94 | else |
| | 95 | true; |
| | 96 | |
| | 97 | if (bits <= int.bits and has_feature) { |
| | 98 | return int.bits; |
| | 99 | } |
| | 100 | } |
| | 101 | |
| | 102 | return null; |
| | 103 | } |
| | 104 | |
| 66 | pub fn getOrGenType(self: *DeclGen, t: Type) !u32 { | 105 | pub fn getOrGenType(self: *DeclGen, t: Type) !u32 { |
| 67 | // We can't use getOrPut here so we can recursively generate types. | 106 | // We can't use getOrPut here so we can recursively generate types. |
| 68 | if (self.types.get(t)) |already_generated| { | 107 | if (self.types.get(t)) |already_generated| { |
| ... | @@ -76,10 +115,12 @@ pub const DeclGen = struct { | ... | @@ -76,10 +115,12 @@ pub const DeclGen = struct { |
| 76 | .Bool => try writeInstruction(&self.spv.types_and_globals, .OpTypeBool, &[_]u32{ result }), | 115 | .Bool => try writeInstruction(&self.spv.types_and_globals, .OpTypeBool, &[_]u32{ result }), |
| 77 | .Int => { | 116 | .Int => { |
| 78 | const int_info = t.intInfo(self.module.getTarget()); | 117 | const int_info = t.intInfo(self.module.getTarget()); |
| 79 | // TODO: Capabilities. | 118 | const backing_bits = self.backingIntBits(int_info.bits) orelse |
| | 119 | return self.fail(.{.node_offset = 0}, "TODO: SPIR-V backend: implement fallback for integer of {} bits", .{ int_info.bits }); |
| | 120 | |
| 80 | try writeInstruction(&self.spv.types_and_globals, .OpTypeInt, &[_]u32{ | 121 | try writeInstruction(&self.spv.types_and_globals, .OpTypeInt, &[_]u32{ |
| 81 | result, | 122 | result, |
| 82 | int_info.bits, | 123 | backing_bits, |
| 83 | switch (int_info.signedness) { | 124 | switch (int_info.signedness) { |
| 84 | .unsigned => 0, | 125 | .unsigned => 0, |
| 85 | .signed => 1, | 126 | .signed => 1, |
Robin Voetter <robin@voetter.nl> 2021-05-15 01:38:39+02:00
src