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
|
; RUN: opt -S -dxil-intrinsic-expansion -mtriple=dxil-pc-shadermodel6.3-library %s | FileCheck %s --check-prefixes=CHECK,EXPCHECK
; RUN: opt -S -dxil-intrinsic-expansion -dxil-op-lower -mtriple=dxil-pc-shadermodel6.3-library %s | FileCheck %s --check-prefixes=CHECK,DOPCHECK
; Make sure dxil operation function calls for pow are generated.
define noundef float @pow_float(float noundef %a, float noundef %b) {
entry:
; DOPCHECK: call float @dx.op.unary.f32(i32 23, float %a)
; EXPCHECK: call float @llvm.log2.f32(float %a)
; CHECK: fmul float %{{.*}}, %b
; DOPCHECK: call float @dx.op.unary.f32(i32 21, float %{{.*}})
; EXPCHECK: call float @llvm.exp2.f32(float %{{.*}})
%elt.pow = call float @llvm.pow.f32(float %a, float %b)
ret float %elt.pow
}
define noundef half @pow_half(half noundef %a, half noundef %b) {
entry:
; DOPCHECK: call half @dx.op.unary.f16(i32 23, half %a)
; EXPCHECK: call half @llvm.log2.f16(half %a)
; CHECK: fmul half %{{.*}}, %b
; DOPCHECK: call half @dx.op.unary.f16(i32 21, half %{{.*}})
; EXPCHECK: call half @llvm.exp2.f16(half %{{.*}})
%elt.pow = call half @llvm.pow.f16(half %a, half %b)
ret half %elt.pow
}
define noundef float @powi_float(float noundef %a, i32 noundef %b) {
entry:
; CHECK: [[CAST:%.*]] = sitofp i32 %b to float
; DOPCHECK: call float @dx.op.unary.f32(i32 23, float %a)
; EXPCHECK: call float @llvm.log2.f32(float %a)
; CHECK: fmul float %{{.*}}, [[CAST]]
; DOPCHECK: call float @dx.op.unary.f32(i32 21, float %{{.*}})
; EXPCHECK: call float @llvm.exp2.f32(float %{{.*}})
%elt.powi = call float @llvm.powi.f32.i32(float %a, i32 %b)
ret float %elt.powi
}
define noundef half @powi_half(half noundef %a, i32 noundef %b) {
entry:
; CHECK: [[CAST:%.*]] = sitofp i32 %b to half
; DOPCHECK: call half @dx.op.unary.f16(i32 23, half %a)
; EXPCHECK: call half @llvm.log2.f16(half %a)
; CHECK: fmul half %{{.*}}, [[CAST]]
; DOPCHECK: call half @dx.op.unary.f16(i32 21, half %{{.*}})
; EXPCHECK: call half @llvm.exp2.f16(half %{{.*}})
%elt.powi = call half @llvm.powi.f16.i32(half %a, i32 %b)
ret half %elt.powi
}
declare half @llvm.pow.f16(half,half)
declare float @llvm.pow.f32(float,float)
declare half @llvm.powi.f16.i32(half,i32)
declare float @llvm.powi.f32.i32(float,i32)
|
