writer/spriv: Cast level parameter to f32 for depth texture sampling

SPIR-V expects a f32, while WGSL has an i32 type for these.

Bug: tint:143
Change-Id: I8e724f24f12154f57cddfd01dfd727e440ca8450
Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/37449
Commit-Queue: Ben Clayton <bclayton@google.com>
Reviewed-by: David Neto <dneto@google.com>

src/writer/spirv/builder.cc

diff --git a/src/writer/spirv/builder.cc b/src/writer/spirv/builder.cc
index c8a887b..1bdd166 100644
--- a/src/writer/spirv/builder.cc
+++ b/src/writer/spirv/builder.cc
@@ -2257,8 +2257,21 @@
         return false;
       }
       assert(pidx.level != kNotUsed);
-      image_operands.emplace_back(
-          ImageOperand{SpvImageOperandsLodMask, gen_param(pidx.level)});
+      auto level = Operand::Int(0);
+      if (call->params()[pidx.level]->result_type()->Is<ast::type::I32>()) {
+        // Depth textures have i32 parameters for the level, but SPIR-V expects
+        // F32. Cast.
+        auto* f32 = mod_->create<ast::type::F32>();
+        ast::TypeConstructorExpression cast(Source{}, f32,
+                                            {call->params()[pidx.level]});
+        level = Operand::Int(GenerateExpression(&cast));
+        if (level.to_i() == 0) {
+          return false;
+        }
+      } else {
+        level = gen_param(pidx.level);
+      }
+      image_operands.emplace_back(ImageOperand{SpvImageOperandsLodMask, level});
       break;
     }
     case ast::Intrinsic::kTextureSampleGrad: {

src/writer/spirv/builder_intrinsic_texture_test.cc

diff --git a/src/writer/spirv/builder_intrinsic_texture_test.cc b/src/writer/spirv/builder_intrinsic_texture_test.cc
index a2ff48a..c4379a4 100644
--- a/src/writer/spirv/builder_intrinsic_texture_test.cc
+++ b/src/writer/spirv/builder_intrinsic_texture_test.cc
@@ -1679,14 +1679,15 @@
 %16 = OpConstant %4 1
 %17 = OpConstant %4 2
 %18 = OpConstantComposite %15 %16 %17
-%19 = OpTypeInt 32 1
-%20 = OpConstant %19 3
+%20 = OpTypeInt 32 1
+%21 = OpConstant %20 3
 )",
           R"(
 %11 = OpLoad %7 %5
 %12 = OpLoad %3 %1
 %14 = OpSampledImage %13 %12 %11
-%9 = OpImageSampleExplicitLod %10 %14 %18 Lod %20
+%19 = OpConvertSToF %4 %21
+%9 = OpImageSampleExplicitLod %10 %14 %18 Lod %19
 %8 = OpCompositeExtract %4 %9 0
 )",
           R"(
@@ -1707,18 +1708,19 @@
 %16 = OpConstant %4 1
 %17 = OpConstant %4 2
 %18 = OpConstantComposite %15 %16 %17
-%19 = OpTypeInt 32 1
-%20 = OpConstant %19 3
-%21 = OpTypeVector %19 2
-%22 = OpConstant %19 4
-%23 = OpConstant %19 5
-%24 = OpConstantComposite %21 %22 %23
+%20 = OpTypeInt 32 1
+%21 = OpConstant %20 3
+%22 = OpTypeVector %20 2
+%23 = OpConstant %20 4
+%24 = OpConstant %20 5
+%25 = OpConstantComposite %22 %23 %24
 )",
           R"(
 %11 = OpLoad %7 %5
 %12 = OpLoad %3 %1
 %14 = OpSampledImage %13 %12 %11
-%9 = OpImageSampleExplicitLod %10 %14 %18 Lod|ConstOffset %20 %24
+%19 = OpConvertSToF %4 %21
+%9 = OpImageSampleExplicitLod %10 %14 %18 Lod|ConstOffset %19 %25
 %8 = OpCompositeExtract %4 %9 0
 )",
           R"(
@@ -1740,7 +1742,7 @@
 %17 = OpConstant %4 2
 %19 = OpTypeInt 32 1
 %20 = OpConstant %19 3
-%22 = OpConstant %19 4
+%23 = OpConstant %19 4
 )",
           R"(
 %11 = OpLoad %7 %5
@@ -1748,6 +1750,7 @@
 %14 = OpSampledImage %13 %12 %11
 %18 = OpConvertSToF %4 %20
 %21 = OpCompositeConstruct %15 %16 %17 %18
+%22 = OpConvertSToF %4 %23
 %9 = OpImageSampleExplicitLod %10 %14 %21 Lod %22
 %8 = OpCompositeExtract %4 %9 0
 )",
@@ -1770,11 +1773,11 @@
 %17 = OpConstant %4 2
 %19 = OpTypeInt 32 1
 %20 = OpConstant %19 3
-%22 = OpConstant %19 4
-%23 = OpTypeVector %19 2
-%24 = OpConstant %19 5
-%25 = OpConstant %19 6
-%26 = OpConstantComposite %23 %24 %25
+%23 = OpConstant %19 4
+%24 = OpTypeVector %19 2
+%25 = OpConstant %19 5
+%26 = OpConstant %19 6
+%27 = OpConstantComposite %24 %25 %26
 )",
           R"(
 %11 = OpLoad %7 %5
@@ -1782,7 +1785,8 @@
 %14 = OpSampledImage %13 %12 %11
 %18 = OpConvertSToF %4 %20
 %21 = OpCompositeConstruct %15 %16 %17 %18
-%9 = OpImageSampleExplicitLod %10 %14 %21 Lod|ConstOffset %22 %26
+%22 = OpConvertSToF %4 %23
+%9 = OpImageSampleExplicitLod %10 %14 %21 Lod|ConstOffset %22 %27
 %8 = OpCompositeExtract %4 %9 0
 )",
           R"(
@@ -1804,14 +1808,15 @@
 %17 = OpConstant %4 2
 %18 = OpConstant %4 3
 %19 = OpConstantComposite %15 %16 %17 %18
-%20 = OpTypeInt 32 1
-%21 = OpConstant %20 4
+%21 = OpTypeInt 32 1
+%22 = OpConstant %21 4
 )",
           R"(
 %11 = OpLoad %7 %5
 %12 = OpLoad %3 %1
 %14 = OpSampledImage %13 %12 %11
-%9 = OpImageSampleExplicitLod %10 %14 %19 Lod %21
+%20 = OpConvertSToF %4 %22
+%9 = OpImageSampleExplicitLod %10 %14 %19 Lod %20
 %8 = OpCompositeExtract %4 %9 0
 )",
           R"(
@@ -1833,7 +1838,7 @@
 %17 = OpConstant %4 3
 %19 = OpTypeInt 32 1
 %20 = OpConstant %19 4
-%22 = OpConstant %19 5
+%23 = OpConstant %19 5
 )",
           R"(
 %11 = OpLoad %7 %5
@@ -1841,6 +1846,7 @@
 %14 = OpSampledImage %13 %12 %11
 %18 = OpConvertSToF %4 %20
 %21 = OpCompositeConstruct %10 %15 %16 %17 %18
+%22 = OpConvertSToF %4 %23
 %9 = OpImageSampleExplicitLod %10 %14 %21 Lod %22
 %8 = OpCompositeExtract %4 %9 0
 )",