641de84d3b
fixes #135221 - have powi use the same legalization path as pow - use CreateSIToFP to cast the int back to a float type - add tests for powi
56 lines
2.2 KiB
LLVM
56 lines
2.2 KiB
LLVM
; 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)
|