[Flint] adding array_length and some math operations encoding

2026-08-30 19:57:09 +02:00
parent cd4f87a99f
commit d357b722ba
18 changed files with 622 additions and 109 deletions
@@ -68,6 +68,10 @@ fn reserveExistingPhysicalRegisters(program: *const program_ir.Program, initial:
                reserveRegister(&next_byte, op.byte_offset.register, grf_size);
                reserveRegister(&next_byte, op.source.register, grf_size);
            },
+            .array_length => |op| {
+                reserveRegister(&next_byte, op.destination.register, grf_size);
+                reserveRegister(&next_byte, op.byte_offset.register, grf_size);
+            },
            .surface_read => |op| {
                reserveRegister(&next_byte, op.destination.register, grf_size);
                reserveRegister(&next_byte, op.address.register, grf_size);
@@ -94,6 +98,11 @@ fn reserveExistingPhysicalRegisters(program: *const program_ir.Program, initial:
                reserveRegister(&next_byte, op.lhs.register, grf_size);
                reserveRegister(&next_byte, op.rhs.register, grf_size);
            },
+            .math => |op| {
+                reserveRegister(&next_byte, op.destination.register, grf_size);
+                reserveRegister(&next_byte, op.lhs.register, grf_size);
+                reserveRegister(&next_byte, op.rhs.register, grf_size);
+            },
            .parallel_copy => return Error.ParallelCopiesNotLowered,
        }
    }
@@ -125,6 +134,10 @@ fn rewriteProgram(program: *program_ir.Program, allocations: []const ?operand.Ph
                try rewriteSource(program, &op.byte_offset, allocations);
                try rewriteSource(program, &op.source, allocations);
            },
+            .array_length => |*op| {
+                try rewriteDestination(program, &op.destination, allocations);
+                try rewriteSource(program, &op.byte_offset, allocations);
+            },
            .surface_read => |*op| {
                try rewriteDestination(program, &op.destination, allocations);
                try rewriteSource(program, &op.address, allocations);
@@ -151,6 +164,11 @@ fn rewriteProgram(program: *program_ir.Program, allocations: []const ?operand.Ph
                try rewriteSource(program, &op.lhs, allocations);
                try rewriteSource(program, &op.rhs, allocations);
            },
+            .math => |*op| {
+                try rewriteDestination(program, &op.destination, allocations);
+                try rewriteSource(program, &op.lhs, allocations);
+                try rewriteSource(program, &op.rhs, allocations);
+            },
            .parallel_copy => return Error.ParallelCopiesNotLowered,
        }
    }