test/CodeGen/SPIRV/hlsl-intrinsics/WaveReadLaneFirst.ll

; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv1.6-unknown-vulkan1.3-compute %s -o - | FileCheck %s
; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv1.6-unknown-vulkan1.3-compute %s -o - -filetype=obj | spirv-val --target-env vulkan1.3 %}

; Test WaveReadLaneFirst lowering for scalar and vector types.

; CHECK: Capability Shader
; CHECK: Capability GroupNonUniformBallot

; CHECK-DAG: %[[#uint:]] = OpTypeInt 32 0
; CHECK-DAG: %[[#f32:]] = OpTypeFloat 32
; CHECK-DAG: %[[#v4_float:]] = OpTypeVector %[[#f32]] 4
; CHECK-DAG: %[[#bool:]] = OpTypeBool
; CHECK-DAG: %[[#scope:]] = OpConstant %[[#uint]] 3

; CHECK-LABEL: Begin function test_float
; CHECK: %[[#fexpr:]] = OpFunctionParameter %[[#f32]]
define internal float @test_float(float %fexpr) {
entry:
; CHECK: %[[#]] = OpGroupNonUniformBroadcastFirst %[[#f32]] %[[#scope]] %[[#fexpr]]
  %0 = call float @llvm.spv.wave.readlane.first.f32(float %fexpr)
  ret float %0
}

; CHECK-LABEL: Begin function test_int
; CHECK: %[[#iexpr:]] = OpFunctionParameter %[[#uint]]
define internal i32 @test_int(i32 %iexpr) {
entry:
; CHECK: %[[#]] = OpGroupNonUniformBroadcastFirst %[[#uint]] %[[#scope]] %[[#iexpr]]
  %0 = call i32 @llvm.spv.wave.readlane.first.i32(i32 %iexpr)
  ret i32 %0
}

; CHECK-LABEL: Begin function test_bool
; CHECK: %[[#bexpr:]] = OpFunctionParameter %[[#bool]]
define internal i1 @test_bool(i1 %bexpr) {
entry:
; CHECK: %[[#]] = OpGroupNonUniformBroadcastFirst %[[#bool]] %[[#scope]] %[[#bexpr]]
  %0 = call i1 @llvm.spv.wave.readlane.first.i1(i1 %bexpr)
  ret i1 %0
}

; CHECK-LABEL: Begin function test_vfloat
; CHECK: %[[#vfexpr:]] = OpFunctionParameter %[[#v4_float]]
define internal <4 x float> @test_vfloat(<4 x float> %vfexpr) {
entry:
; CHECK: %[[#]] = OpGroupNonUniformBroadcastFirst %[[#v4_float]] %[[#scope]] %[[#vfexpr]]
  %0 = call <4 x float> @llvm.spv.wave.readlane.first.v4f32(
      <4 x float> %vfexpr)
  ret <4 x float> %0
}

define void @main() #0 {
  ret void
}

declare float @llvm.spv.wave.readlane.first.f32(float)
declare i32 @llvm.spv.wave.readlane.first.i32(i32)
declare i1 @llvm.spv.wave.readlane.first.i1(i1)
declare <4 x float> @llvm.spv.wave.readlane.first.v4f32(<4 x float>)

attributes #0 = { "hlsl.numthreads"="1,1,1" "hlsl.shader"="compute" }