From d8b591766ac37090ae77f7b67d0ce6b54c64254e Mon Sep 17 00:00:00 2001
From: Ali Chraghi <alichraghi@proton.me>
Date: Sun, 8 Oct 2023 16:22:37 +0330
Subject: [PATCH] spirv: `fpext` and `fptrunc` instructions

---
 src/codegen/spirv.zig  | 18 ++++++++++++++++++
 src/link/SpirV.zig     |  2 +-
 test/behavior/cast.zig |  1 -
 3 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/codegen/spirv.zig b/src/codegen/spirv.zig
index 7528c40cca..6881b2c598 100644
--- a/src/codegen/spirv.zig
+++ b/src/codegen/spirv.zig
@@ -1957,6 +1957,7 @@ const DeclGen = struct {
             .int_from_ptr    => try self.airIntFromPtr(inst),
             .float_from_int  => try self.airFloatFromInt(inst),
             .int_from_float  => try self.airIntFromFloat(inst),
+            .fpext, .fptrunc => try self.airFloatCast(inst),
             .not             => try self.airNot(inst),
 
             .array_to_slice => try self.airArrayToSlice(inst),
@@ -2685,6 +2686,23 @@ const DeclGen = struct {
         return result_id;
     }
 
+    fn airFloatCast(self: *DeclGen, inst: Air.Inst.Index) !?IdRef {
+        if (self.liveness.isUnused(inst)) return null;
+
+        const ty_op = self.air.instructions.items(.data)[inst].ty_op;
+        const operand_id = try self.resolve(ty_op.operand);
+        const dest_ty = self.typeOfIndex(inst);
+        const dest_ty_id = try self.resolveTypeId(dest_ty);
+
+        const result_id = self.spv.allocId();
+        try self.func.body.emit(self.spv.gpa, .OpFConvert, .{
+            .id_result_type = dest_ty_id,
+            .id_result = result_id,
+            .float_value = operand_id,
+        });
+        return result_id;
+    }
+
     fn airNot(self: *DeclGen, inst: Air.Inst.Index) !?IdRef {
         if (self.liveness.isUnused(inst)) return null;
         const ty_op = self.air.instructions.items(.data)[inst].ty_op;
diff --git a/src/link/SpirV.zig b/src/link/SpirV.zig
index 58d8ca6841..325bb30fe0 100644
--- a/src/link/SpirV.zig
+++ b/src/link/SpirV.zig
@@ -230,7 +230,7 @@ fn writeMemoryModel(spv: *SpvModule, target: std.Target) !void {
     };
 
     // TODO: Put this in a proper section.
-    try spv.sections.capabilities.emit(spv.gpa, .OpMemoryModel, .{
+    try spv.sections.extensions.emit(spv.gpa, .OpMemoryModel, .{
         .addressing_model = addressing_model,
         .memory_model = memory_model,
     });
diff --git a/test/behavior/cast.zig b/test/behavior/cast.zig
index 388d6f4750..4a6b4db2ce 100644
--- a/test/behavior/cast.zig
+++ b/test/behavior/cast.zig
@@ -648,7 +648,6 @@ test "@floatCast cast down" {
     if (builtin.zig_backend == .stage2_aarch64) return error.SkipZigTest; // TODO
     if (builtin.zig_backend == .stage2_arm) return error.SkipZigTest; // TODO
     if (builtin.zig_backend == .stage2_sparc64) return error.SkipZigTest; // TODO
-    if (builtin.zig_backend == .stage2_spirv64) return error.SkipZigTest;
 
     {
         var double: f64 = 0.001534;