1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv1.5-vulkan-unknown %s -o - | FileCheck %s
; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv1.5-vulkan-unknown %s -o - -filetype=obj | spirv-val %}
; Test lowering to spir-v backend for various types and scalar/vector
; CHECK: Capability Shader
; CHECK: Capability GroupNonUniformShuffle
; CHECK-DAG: %[[#uint:]] = OpTypeInt 32 0
; CHECK-DAG: %[[#f32:]] = OpTypeFloat 32
; CHECK-DAG: %[[#v4_float:]] = OpTypeVector %[[#f32]] 4
; CHECK-DAG: %[[#bool:]] = OpTypeBool
; CHECK-DAG: %[[#v4_bool:]] = OpTypeVector %[[#bool]] 4
; CHECK-DAG: %[[#scope:]] = OpConstant %[[#uint]] 3
; CHECK-LABEL: Begin function test_float
; CHECK: %[[#fexpr:]] = OpFunctionParameter %[[#f32]]
; CHECK: %[[#idx1:]] = OpFunctionParameter %[[#uint]]
define float @test_float(float %fexpr, i32 %idx) {
entry:
; CHECK: %[[#fret:]] = OpGroupNonUniformShuffle %[[#f32]] %[[#scope]] %[[#fexpr]] %[[#idx1]]
%0 = call float @llvm.spv.wave.readlane.f32(float %fexpr, i32 %idx)
ret float %0
}
; CHECK-LABEL: Begin function test_int
; CHECK: %[[#iexpr:]] = OpFunctionParameter %[[#uint]]
; CHECK: %[[#idx2:]] = OpFunctionParameter %[[#uint]]
define i32 @test_int(i32 %iexpr, i32 %idx) {
entry:
; CHECK: %[[#iret:]] = OpGroupNonUniformShuffle %[[#uint]] %[[#scope]] %[[#iexpr]] %[[#idx2]]
%0 = call i32 @llvm.spv.wave.readlane.i32(i32 %iexpr, i32 %idx)
ret i32 %0
}
; CHECK-LABEL: Begin function test_vbool
; CHECK: %[[#vbexpr:]] = OpFunctionParameter %[[#v4_bool]]
; CHECK: %[[#idx3:]] = OpFunctionParameter %[[#uint]]
define <4 x i1> @test_vbool(<4 x i1> %vbexpr, i32 %idx) {
entry:
; CHECK: %[[#vbret:]] = OpGroupNonUniformShuffle %[[#v4_bool]] %[[#scope]] %[[#vbexpr]] %[[#idx3]]
%0 = call <4 x i1> @llvm.spv.wave.readlane.v4i1(<4 x i1> %vbexpr, i32 %idx)
ret <4 x i1> %0
}
; CHECK-LABEL: Begin function test_vfloat
; CHECK: %[[#vfexpr:]] = OpFunctionParameter %[[#v4_float]]
; CHECK: %[[#idx4:]] = OpFunctionParameter %[[#uint]]
define <4 x float> @test_vfloat(<4 x float> %vfexpr, i32 %idx) {
entry:
; CHECK: %[[#vbret:]] = OpGroupNonUniformShuffle %[[#v4_float]] %[[#scope]] %[[#vfexpr]] %[[#idx4]]
%0 = call <4 x float> @llvm.spv.wave.readlane.v4f32(<4 x float> %vfexpr, i32 %idx)
ret <4 x float> %0
}
declare float @llvm.spv.wave.readlane.f32(float, i32)
declare i32 @llvm.spv.wave.readlane.i32(i32, i32)
declare <4 x i1> @llvm.spv.wave.readlane.v4i1(<4 x i1>, i32)
declare <4 x float> @llvm.spv.wave.readlane.v4f32(<4 x float>, i32)
|
