spirv: support enum integer values in Assembler

Snektron · Feb 3, 2024 · 17442d3 · 17442d3
1 parent 9481f57
commit 17442d3
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/codegen/spirv/Assembler.zig b/src/codegen/spirv/Assembler.zig
@@ -591,9 +591,13 @@ fn parseValueEnum(self: *Assembler, kind: spec.OperandKind) !void {
     try self.expectToken(.value);
 
     const text = self.tokenText(tok);
+    const int_value = std.fmt.parseInt(u32, text, 0) catch null;
     const enumerant = for (kind.enumerants()) |enumerant| {
-        if (std.mem.eql(u8, enumerant.name, text))
-            break enumerant;
+        if (int_value) |v| {
+            if (v == enumerant.value) break enumerant;
+        } else {
+            if (std.mem.eql(u8, enumerant.name, text)) break enumerant;
+        }
     } else {
         return self.fail(tok.start, "'{s}' is not a valid value for enumeration {s}", .{ text, @tagName(kind) });
     };