Skip to content

Commit b5bdcc7

Browse files
committed
Add TraceRay SROA pass test
1 parent c197679 commit b5bdcc7

1 file changed

Lines changed: 180 additions & 0 deletions

File tree

Lines changed: 180 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,180 @@
1+
; RUN: %dxopt %s -hlsl-passes-resume -scalarrepl-param-hlsl -S | FileCheck %s
2+
3+
; CHECK: define <4 x float> @"
4+
; CHECK-SAME: ?emit@@YA?AV?$vector@M$03@@AIAV?$vector@M$01@@URayDesc@@UPayload@@@Z"(<2 x float>* noalias dereferenceable(8) %f2, %struct.RayDesc* %Ray, %struct.Payload* noalias %p)
5+
6+
; Copy Payload fields (PLD_F0, PLD_F1) to local allocas:
7+
; CHECK: %[[GEP:[^ ,]+]] = getelementptr inbounds %struct.Payload, %struct.Payload* %p, i32 0, i32 0
8+
; CHECK: %[[LOAD:[^ ,]+]] = load <2 x float>, <2 x float>* %[[GEP]]
9+
; CHECK: store <2 x float> %[[LOAD]], <2 x float>* %[[PLD_F0:[^ ,]+]]
10+
; CHECK: %[[GEP:[^ ,]+]] = getelementptr inbounds %struct.Payload, %struct.Payload* %p, i32 0, i32 1
11+
; CHECK: %[[LOAD:[^ ,]+]] = load <3 x i32>, <3 x i32>* %[[GEP]]
12+
; CHECK: store <3 x i32> %[[LOAD]], <3 x i32>* %[[PLD_F1:[^ ,]+]]
13+
14+
; CHECK: %[[RTAS:[^ ,]+]] = call %dx.types.Handle @"dx.hl.annotatehandle..%dx.types.Handle (i32, %dx.types.Handle, %dx.types.ResourceProperties, %struct.RaytracingAccelerationStructure)"(i32 14, %dx.types.Handle %21, %dx.types.ResourceProperties { i32 16, i32 0 }, %struct.RaytracingAccelerationStructure undef)
15+
16+
; Copy RayDesc fields (Origin, TMin, Direction, TMax) to local allocas:
17+
; CHECK: %[[RAY_ORIGIN_GEP:[^ ,]+]] = getelementptr inbounds %struct.RayDesc, %struct.RayDesc* %Ray, i32 0, i32 0
18+
; CHECK: %[[RAY_ORIGIN_LOAD:[^ ,]+]] = load <3 x float>, <3 x float>* %[[RAY_ORIGIN_GEP]]
19+
; CHECK: store <3 x float> %[[RAY_ORIGIN_LOAD]], <3 x float>* %[[RAY_ORIGIN_P0:[^ ,]+]]
20+
; CHECK: %[[TMIN_GEP:[^ ,]+]] = getelementptr inbounds %struct.RayDesc, %struct.RayDesc* %Ray, i32 0, i32 1
21+
; CHECK: %[[TMIN_LOAD:[^ ,]+]] = load float, float* %[[TMIN_GEP]]
22+
; CHECK: store float %[[TMIN_LOAD]], float* %[[TMIN_P0:[^ ,]+]]
23+
; CHECK: %[[DIRECTION_GEP:[^ ,]+]] = getelementptr inbounds %struct.RayDesc, %struct.RayDesc* %Ray, i32 0, i32 2
24+
; CHECK: %[[DIRECTION_LOAD:[^ ,]+]] = load <3 x float>, <3 x float>* %[[DIRECTION_GEP]]
25+
; CHECK: store <3 x float> %[[DIRECTION_LOAD]], <3 x float>* %[[DIRECTION_P0:[^ ,]+]]
26+
; CHECK: %[[TMAX_GEP:[^ ,]+]] = getelementptr inbounds %struct.RayDesc, %struct.RayDesc* %Ray, i32 0, i32 3
27+
; CHECK: %[[TMAX_LOAD:[^ ,]+]] = load float, float* %[[TMAX_GEP]]
28+
; CHECK: store float %[[TMAX_LOAD]], float* %[[TMAX_P0:[^ ,]+]]
29+
30+
; COM: Copy Payload fields into payload struct for call:
31+
; CHECK: %[[PLD_F0_GEP:[^ ,]+]] = getelementptr inbounds %struct.Payload, %struct.Payload* %[[PLD_P0:[^ ,]+]], i32 0, i32 0
32+
; CHECK: %[[PLD_F0_LOAD:[^ ,]+]] = load <2 x float>, <2 x float>* %[[PLD_F0]]
33+
; CHECK: store <2 x float> %[[PLD_F0_LOAD]], <2 x float>* %[[PLD_F0_GEP]]
34+
; CHECK: %[[PLD_F1_GEP:[^ ,]+]] = getelementptr inbounds %struct.Payload, %struct.Payload* %[[PLD_P0]], i32 0, i32 1
35+
; CHECK: %[[PLD_F1_LOAD:[^ ,]+]] = load <3 x i32>, <3 x i32>* %[[PLD_F1]]
36+
; CHECK: store <3 x i32> %[[PLD_F1_LOAD]], <3 x i32>* %[[PLD_F1_GEP]]
37+
38+
; COM: Load RayDesc fields:
39+
; CHECK: %[[RAY_ORIGIN_LOAD2:[^ ,]+]] = load <3 x float>, <3 x float>* %[[RAY_ORIGIN_P0]]
40+
; CHECK: %[[TMIN_LOAD2:[^ ,]+]] = load float, float* %[[TMIN_P0]]
41+
; CHECK: %[[DIRECTION_LOAD2:[^ ,]+]] = load <3 x float>, <3 x float>* %[[DIRECTION_P0]]
42+
; CHECK: %[[TMAX_LOAD2:[^ ,]+]] = load float, float* %[[TMAX_P0]]
43+
44+
; call TraceRay with the local allocas:
45+
; CHECK: call void @"dx.hl.op..void (i32, %dx.types.Handle, i32, i32, i32, i32, i32, <3 x float>, float, <3 x float>, float, %struct.Payload*)"(i32 69, %dx.types.Handle %[[RTAS]], i32 %19, i32 %18, i32 %17, i32 %16, i32 %15, <3 x float> %[[RAY_ORIGIN_LOAD2]], float %[[TMIN_LOAD2]], <3 x float> %[[DIRECTION_LOAD2]], float %[[TMAX_LOAD2]], %struct.Payload* %[[PLD_P0]])
46+
47+
48+
target datalayout = "e-m:e-p:32:32-i1:32-i8:32-i16:32-i32:32-i64:64-f16:32-f32:32-f64:64-n8:16:32:64"
49+
target triple = "dxil-ms-dx"
50+
51+
%struct.RaytracingAccelerationStructure = type { i32 }
52+
%"$Globals" = type { i32, i32, i32, i32, i32 }
53+
%struct.RayDesc = type { <3 x float>, float, <3 x float>, float }
54+
%struct.Payload = type { <2 x float>, <3 x i32> }
55+
%dx.types.Handle = type { i8* }
56+
%dx.types.ResourceProperties = type { i32, i32 }
57+
58+
@"\01?Acc@@3URaytracingAccelerationStructure@@A" = external global %struct.RaytracingAccelerationStructure, align 4
59+
@"\01?RayFlags@@3IB" = external constant i32, align 4
60+
@"\01?InstanceInclusionMask@@3IB" = external constant i32, align 4
61+
@"\01?RayContributionToHitGroupIndex@@3IB" = external constant i32, align 4
62+
@"\01?MultiplierForGeometryContributionToHitGroupIndex@@3IB" = external constant i32, align 4
63+
@"\01?MissShaderIndex@@3IB" = external constant i32, align 4
64+
@"$Globals" = external constant %"$Globals"
65+
66+
; Function Attrs: nounwind
67+
define <4 x float> @"\01?emit@@YA?AV?$vector@M$03@@AIAV?$vector@M$01@@URayDesc@@UPayload@@@Z"(<2 x float>* noalias dereferenceable(8) %f2, %struct.RayDesc* %Ray, %struct.Payload* noalias %p) #0 {
68+
entry:
69+
%0 = alloca %struct.RayDesc, !dbg !39 ; line:22 col:61
70+
%1 = bitcast %struct.RayDesc* %0 to i8*, !dbg !39 ; line:22 col:61
71+
%2 = bitcast %struct.RayDesc* %Ray to i8*, !dbg !39 ; line:22 col:61
72+
call void @llvm.memcpy.p0i8.p0i8.i64(i8* %1, i8* %2, i64 32, i32 1, i1 false), !dbg !39 ; line:22 col:61
73+
%3 = call %dx.types.Handle @"dx.hl.createhandle..%dx.types.Handle (i32, %\22$Globals\22*, i32)"(i32 0, %"$Globals"* @"$Globals", i32 0), !dbg !39 ; line:22 col:61
74+
%4 = call %dx.types.Handle @"dx.hl.annotatehandle..%dx.types.Handle (i32, %dx.types.Handle, %dx.types.ResourceProperties, %\22$Globals\22)"(i32 14, %dx.types.Handle %3, %dx.types.ResourceProperties { i32 13, i32 20 }, %"$Globals" undef), !dbg !39 ; line:22 col:61
75+
%5 = call %"$Globals"* @"dx.hl.subscript.cb.rn.%\22$Globals\22* (i32, %dx.types.Handle, i32)"(i32 6, %dx.types.Handle %4, i32 0), !dbg !39 ; line:22 col:61
76+
%6 = getelementptr inbounds %"$Globals", %"$Globals"* %5, i32 0, i32 0, !dbg !39 ; line:22 col:61
77+
%7 = getelementptr inbounds %"$Globals", %"$Globals"* %5, i32 0, i32 1, !dbg !39 ; line:22 col:61
78+
%8 = getelementptr inbounds %"$Globals", %"$Globals"* %5, i32 0, i32 2, !dbg !39 ; line:22 col:61
79+
%9 = getelementptr inbounds %"$Globals", %"$Globals"* %5, i32 0, i32 3, !dbg !39 ; line:22 col:61
80+
%10 = getelementptr inbounds %"$Globals", %"$Globals"* %5, i32 0, i32 4, !dbg !39 ; line:22 col:61
81+
%11 = load i32, i32* %10, align 4, !dbg !39, !tbaa !43 ; line:22 col:61
82+
%12 = load i32, i32* %9, align 4, !dbg !47, !tbaa !43 ; line:22 col:12
83+
%13 = load i32, i32* %8, align 4, !dbg !48, !tbaa !43 ; line:21 col:12
84+
%14 = load i32, i32* %7, align 4, !dbg !49, !tbaa !43 ; line:20 col:25
85+
%15 = load i32, i32* %6, align 4, !dbg !50, !tbaa !43 ; line:20 col:16
86+
%16 = load %struct.RaytracingAccelerationStructure, %struct.RaytracingAccelerationStructure* @"\01?Acc@@3URaytracingAccelerationStructure@@A", !dbg !51 ; line:20 col:3
87+
%17 = call %dx.types.Handle @"dx.hl.createhandle..%dx.types.Handle (i32, %struct.RaytracingAccelerationStructure)"(i32 0, %struct.RaytracingAccelerationStructure %16), !dbg !51 ; line:20 col:3
88+
%18 = call %dx.types.Handle @"dx.hl.annotatehandle..%dx.types.Handle (i32, %dx.types.Handle, %dx.types.ResourceProperties, %struct.RaytracingAccelerationStructure)"(i32 14, %dx.types.Handle %17, %dx.types.ResourceProperties { i32 16, i32 0 }, %struct.RaytracingAccelerationStructure undef), !dbg !51 ; line:20 col:3
89+
call void @"dx.hl.op..void (i32, %dx.types.Handle, i32, i32, i32, i32, i32, %struct.RayDesc*, %struct.Payload*)"(i32 69, %dx.types.Handle %18, i32 %15, i32 %14, i32 %13, i32 %12, i32 %11, %struct.RayDesc* %0, %struct.Payload* %p), !dbg !51 ; line:20 col:3
90+
ret <4 x float> <float 0x4004CCCCC0000000, float 0x4004CCCCC0000000, float 0x4004CCCCC0000000, float 0x4004CCCCC0000000>, !dbg !52 ; line:24 col:4
91+
}
92+
93+
; Function Attrs: nounwind
94+
declare void @"dx.hl.op..void (i32, %dx.types.Handle, i32, i32, i32, i32, i32, %struct.RayDesc*, %struct.Payload*)"(i32, %dx.types.Handle, i32, i32, i32, i32, i32, %struct.RayDesc*, %struct.Payload*) #0
95+
96+
; Function Attrs: nounwind readnone
97+
declare %dx.types.Handle @"dx.hl.createhandle..%dx.types.Handle (i32, %struct.RaytracingAccelerationStructure)"(i32, %struct.RaytracingAccelerationStructure) #1
98+
99+
; Function Attrs: nounwind readnone
100+
declare %dx.types.Handle @"dx.hl.annotatehandle..%dx.types.Handle (i32, %dx.types.Handle, %dx.types.ResourceProperties, %struct.RaytracingAccelerationStructure)"(i32, %dx.types.Handle, %dx.types.ResourceProperties, %struct.RaytracingAccelerationStructure) #1
101+
102+
; Function Attrs: nounwind readnone
103+
declare %"$Globals"* @"dx.hl.subscript.cb.rn.%\22$Globals\22* (i32, %dx.types.Handle, i32)"(i32, %dx.types.Handle, i32) #1
104+
105+
; Function Attrs: nounwind readnone
106+
declare %dx.types.Handle @"dx.hl.createhandle..%dx.types.Handle (i32, %\22$Globals\22*, i32)"(i32, %"$Globals"*, i32) #1
107+
108+
; Function Attrs: nounwind readnone
109+
declare %dx.types.Handle @"dx.hl.annotatehandle..%dx.types.Handle (i32, %dx.types.Handle, %dx.types.ResourceProperties, %\22$Globals\22)"(i32, %dx.types.Handle, %dx.types.ResourceProperties, %"$Globals") #1
110+
111+
; Function Attrs: nounwind
112+
declare void @llvm.memcpy.p0i8.p0i8.i64(i8* nocapture, i8* nocapture readonly, i64, i32, i1) #0
113+
114+
attributes #0 = { nounwind }
115+
attributes #1 = { nounwind readnone }
116+
117+
!llvm.module.flags = !{!0}
118+
!pauseresume = !{!1}
119+
!llvm.ident = !{!2}
120+
!dx.version = !{!3}
121+
!dx.valver = !{!4}
122+
!dx.shaderModel = !{!5}
123+
!dx.typeAnnotations = !{!6, !21}
124+
!dx.entryPoints = !{!30}
125+
!dx.fnprops = !{}
126+
!dx.options = !{!37, !38}
127+
128+
!0 = !{i32 2, !"Debug Info Version", i32 3}
129+
!1 = !{!"hlsl-hlemit", !"hlsl-hlensure"}
130+
!2 = !{!"dxc(private) 1.8.0.4928 (ser_hlslattributes_patch, 937c16cc6)"}
131+
!3 = !{i32 1, i32 3}
132+
!4 = !{i32 1, i32 9}
133+
!5 = !{!"lib", i32 6, i32 3}
134+
!6 = !{i32 0, %struct.RayDesc undef, !7, %struct.Payload undef, !12, %"$Globals" undef, !15}
135+
!7 = !{i32 32, !8, !9, !10, !11}
136+
!8 = !{i32 6, !"Origin", i32 3, i32 0, i32 7, i32 9}
137+
!9 = !{i32 6, !"TMin", i32 3, i32 12, i32 7, i32 9}
138+
!10 = !{i32 6, !"Direction", i32 3, i32 16, i32 7, i32 9}
139+
!11 = !{i32 6, !"TMax", i32 3, i32 28, i32 7, i32 9}
140+
!12 = !{i32 28, !13, !14}
141+
!13 = !{i32 6, !"t", i32 3, i32 0, i32 7, i32 9}
142+
!14 = !{i32 6, !"t2", i32 3, i32 16, i32 7, i32 4}
143+
!15 = !{i32 20, !16, !17, !18, !19, !20}
144+
!16 = !{i32 6, !"RayFlags", i32 3, i32 0, i32 7, i32 5}
145+
!17 = !{i32 6, !"InstanceInclusionMask", i32 3, i32 4, i32 7, i32 5}
146+
!18 = !{i32 6, !"RayContributionToHitGroupIndex", i32 3, i32 8, i32 7, i32 5}
147+
!19 = !{i32 6, !"MultiplierForGeometryContributionToHitGroupIndex", i32 3, i32 12, i32 7, i32 5}
148+
!20 = !{i32 6, !"MissShaderIndex", i32 3, i32 16, i32 7, i32 5}
149+
!21 = !{i32 1, <4 x float> (<2 x float>*, %struct.RayDesc*, %struct.Payload*)* @"\01?emit@@YA?AV?$vector@M$03@@AIAV?$vector@M$01@@URayDesc@@UPayload@@@Z", !22}
150+
!22 = !{!23, !26, !27, !29}
151+
!23 = !{i32 1, !24, !25}
152+
!24 = !{i32 7, i32 9}
153+
!25 = !{}
154+
!26 = !{i32 2, !24, !25}
155+
!27 = !{i32 0, !28, !25}
156+
!28 = !{i32 4, !"R"}
157+
!29 = !{i32 2, !25, !25}
158+
!30 = !{null, !"", null, !31, null}
159+
!31 = !{!32, null, !35, null}
160+
!32 = !{!33}
161+
!33 = !{i32 0, %struct.RaytracingAccelerationStructure* @"\01?Acc@@3URaytracingAccelerationStructure@@A", !"Acc", i32 -1, i32 -1, i32 1, i32 16, i32 0, !34}
162+
!34 = !{i32 0, i32 4}
163+
!35 = !{!36}
164+
!36 = !{i32 0, %"$Globals"* @"$Globals", !"$Globals", i32 0, i32 -1, i32 1, i32 20, null}
165+
!37 = !{i32 -2147483584}
166+
!38 = !{i32 11}
167+
!39 = !DILocation(line: 22, column: 61, scope: !40)
168+
!40 = !DISubprogram(name: "emit", scope: !41, file: !41, line: 19, type: !42, isLocal: false, isDefinition: true, scopeLine: 19, flags: DIFlagPrototyped, isOptimized: false, function: <4 x float> (<2 x float>*, %struct.RayDesc*, %struct.Payload*)* @"\01?emit@@YA?AV?$vector@M$03@@AIAV?$vector@M$01@@URayDesc@@UPayload@@@Z")
169+
!41 = !DIFile(filename: "D:\5Cgit\5Cdxc\5Cmain\5Ctools\5Cclang\5Ctest\5CHLSLFileCheck\5Cshader_targets\5Craytracing\5Craytracing_traceray.hlsl", directory: "")
170+
!42 = !DISubroutineType(types: !25)
171+
!43 = !{!44, !44, i64 0}
172+
!44 = !{!"int", !45, i64 0}
173+
!45 = !{!"omnipotent char", !46, i64 0}
174+
!46 = !{!"Simple C/C++ TBAA"}
175+
!47 = !DILocation(line: 22, column: 12, scope: !40)
176+
!48 = !DILocation(line: 21, column: 12, scope: !40)
177+
!49 = !DILocation(line: 20, column: 25, scope: !40)
178+
!50 = !DILocation(line: 20, column: 16, scope: !40)
179+
!51 = !DILocation(line: 20, column: 3, scope: !40)
180+
!52 = !DILocation(line: 24, column: 4, scope: !40)

0 commit comments

Comments
 (0)