1
- ; RUN: opt -S -dxil-op-lower -mtriple=dxil-pc-shadermodel6.3-library %s | FileCheck %s
1
+ ; RUN: opt -S -dxil-intrinsic-expansion -mtriple=dxil-pc-shadermodel6.3-library %s | FileCheck %s --check-prefixes=CHECK,EXPCHECK
2
+ ; RUN: opt -S -dxil-intrinsic-expansion -scalarizer -dxil-op-lower -mtriple=dxil-pc-shadermodel6.3-library %s | FileCheck %s --check-prefixes=CHECK,DOPCHECK
2
3
3
4
; Make sure correct dxil expansions for atan2 are generated for float and half.
4
5
5
6
define noundef float @atan2_float (float noundef %y , float noundef %x ) {
6
7
entry:
7
8
; CHECK: [[DIV:%.+]] = fdiv float %y, %x
8
- ; CHECK: [[TAN:%.+]] = call float @dx.op.unary.f32(i32 17, float [[DIV]])
9
- ; CHECK-DAG: [[ADD_PI:%.+]] = fadd float [[TAN]], 0x400921FB60000000
10
- ; CHECK-DAG: [[SUB_PI:%.+]] = fsub float [[TAN]], 0x400921FB60000000
9
+ ; EXPCHECK: [[ATAN:%.+]] = call float @llvm.atan.f32(float [[DIV]])
10
+ ; DOPCHECK: [[ATAN:%.+]] = call float @dx.op.unary.f32(i32 17, float [[DIV]])
11
+ ; CHECK-DAG: [[ADD_PI:%.+]] = fadd float [[ATAN]], 0x400921FB60000000
12
+ ; CHECK-DAG: [[SUB_PI:%.+]] = fsub float [[ATAN]], 0x400921FB60000000
11
13
; CHECK-DAG: [[X_LT_0:%.+]] = fcmp olt float %x, 0.000000e+00
12
14
; CHECK-DAG: [[X_EQ_0:%.+]] = fcmp oeq float %x, 0.000000e+00
13
15
; CHECK-DAG: [[Y_GE_0:%.+]] = fcmp oge float %y, 0.000000e+00
14
16
; CHECK-DAG: [[Y_LT_0:%.+]] = fcmp olt float %y, 0.000000e+00
15
17
; CHECK: [[XLT0_AND_YGE0:%.+]] = and i1 [[X_LT_0]], [[Y_GE_0]]
16
- ; CHECK: [[SELECT_ADD_PI:%.+]] = select i1 [[XLT0_AND_YGE0]], float [[ADD_PI]], float [[TAN ]]
18
+ ; CHECK: [[SELECT_ADD_PI:%.+]] = select i1 [[XLT0_AND_YGE0]], float [[ADD_PI]], float [[ATAN ]]
17
19
; CHECK: [[XLT0_AND_YLT0:%.+]] = and i1 [[X_LT_0]], [[Y_LT_0]]
18
20
; CHECK: [[SELECT_SUB_PI:%.+]] = select i1 [[XLT0_AND_YLT0]], float [[SUB_PI]], float [[SELECT_ADD_PI]]
19
21
; CHECK: [[XEQ0_AND_YLT0:%.+]] = and i1 [[X_EQ_0]], [[Y_LT_0]]
@@ -28,15 +30,16 @@ entry:
28
30
define noundef half @atan2_half (half noundef %y , half noundef %x ) {
29
31
entry:
30
32
; CHECK: [[DIV:%.+]] = fdiv half %y, %x
31
- ; CHECK: [[TAN:%.+]] = call half @dx.op.unary.f16(i32 17, half [[DIV]])
32
- ; CHECK-DAG: [[ADD_PI:%.+]] = fadd half [[TAN]], 0xH4248
33
- ; CHECK-DAG: [[SUB_PI:%.+]] = fsub half [[TAN]], 0xH4248
33
+ ; EXPCHECK: [[ATAN:%.+]] = call half @llvm.atan.f16(half [[DIV]])
34
+ ; DOPCHECK: [[ATAN:%.+]] = call half @dx.op.unary.f16(i32 17, half [[DIV]])
35
+ ; CHECK-DAG: [[ADD_PI:%.+]] = fadd half [[ATAN]], 0xH4248
36
+ ; CHECK-DAG: [[SUB_PI:%.+]] = fsub half [[ATAN]], 0xH4248
34
37
; CHECK-DAG: [[X_LT_0:%.+]] = fcmp olt half %x, 0xH0000
35
38
; CHECK-DAG: [[X_EQ_0:%.+]] = fcmp oeq half %x, 0xH0000
36
39
; CHECK-DAG: [[Y_GE_0:%.+]] = fcmp oge half %y, 0xH0000
37
40
; CHECK-DAG: [[Y_LT_0:%.+]] = fcmp olt half %y, 0xH0000
38
41
; CHECK: [[XLT0_AND_YGE0:%.+]] = and i1 [[X_LT_0]], [[Y_GE_0]]
39
- ; CHECK: [[SELECT_ADD_PI:%.+]] = select i1 [[XLT0_AND_YGE0]], half [[ADD_PI]], half [[TAN ]]
42
+ ; CHECK: [[SELECT_ADD_PI:%.+]] = select i1 [[XLT0_AND_YGE0]], half [[ADD_PI]], half [[ATAN ]]
40
43
; CHECK: [[XLT0_AND_YLT0:%.+]] = and i1 [[X_LT_0]], [[Y_LT_0]]
41
44
; CHECK: [[SELECT_SUB_PI:%.+]] = select i1 [[XLT0_AND_YLT0]], half [[SUB_PI]], half [[SELECT_ADD_PI]]
42
45
; CHECK: [[XEQ0_AND_YLT0:%.+]] = and i1 [[X_EQ_0]], [[Y_LT_0]]
0 commit comments