7c366b041c
Fixes #137209 This PR: - Adds a case to `expandIntrinsic()` in `DXILIntrinsicExpansion.cpp` to expand the `Intrinsic::is_fpclass` in the case of `FPClassTest::fcNegZero` - Defines the `IsNaN`, `IsFinite`, `IsNormal` DXIL ops in `DXIL.td` - Adds a case to `lowerIntrinsics()` in `DXILOpLowering.cpp` to handle the lowering of `Intrinsic::is_fpclass` to the DXIL ops `IsNaN`, `IsInf`, `IsFinite`, `IsNormal` when the FPClassTest is `fcNan`, `fcInf`, `fcFinite`, and `fcNormal` respectively - Creates a test `llvm/test/CodeGen/DirectX/is_fpclass.ll` to exercise the intrinsic expansion and DXIL op lowering of `Intrinsic::is_fpclass` ~~A separate PR will be made to remove the now-redundant `dx_isinf` intrinsic to address #87777.~~ A proper implementation for the lowering of the `llvm.is.fpclass` intrinsic to handle all possible combinations of FPClassTest can be implemented in a separate PR. This PR's implementation focuses primarily on addressing the current use-cases for DirectML and HLSL intrinsics.
155 lines
6.5 KiB
LLVM
155 lines
6.5 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
|
|
; RUN: opt -S -dxil-intrinsic-expansion -scalarizer -dxil-op-lower -mtriple=dxil-pc-shadermodel6.3-library %s | FileCheck %s
|
|
|
|
|
|
define noundef i1 @isnegzero(float noundef %a) {
|
|
; CHECK-LABEL: define noundef i1 @isnegzero(
|
|
; CHECK-SAME: float noundef [[A:%.*]]) {
|
|
; CHECK-NEXT: [[ENTRY:.*:]]
|
|
; CHECK-NEXT: [[TMP0:%.*]] = bitcast float [[A]] to i32
|
|
; CHECK-NEXT: [[IS_FPCLASS_NEGZERO:%.*]] = icmp eq i32 [[TMP0]], -2147483648
|
|
; CHECK-NEXT: ret i1 [[IS_FPCLASS_NEGZERO]]
|
|
;
|
|
entry:
|
|
%0 = call i1 @llvm.is.fpclass.f32(float %a, i32 32)
|
|
ret i1 %0
|
|
}
|
|
|
|
define noundef <2 x i1> @isnegzerov2(<2 x float> noundef %a) {
|
|
; CHECK-LABEL: define noundef <2 x i1> @isnegzerov2(
|
|
; CHECK-SAME: <2 x float> noundef [[A:%.*]]) {
|
|
; CHECK-NEXT: [[ENTRY:.*:]]
|
|
; CHECK-NEXT: [[A_I0:%.*]] = extractelement <2 x float> [[A]], i64 0
|
|
; CHECK-NEXT: [[DOTI0:%.*]] = bitcast float [[A_I0]] to i32
|
|
; CHECK-NEXT: [[A_I1:%.*]] = extractelement <2 x float> [[A]], i64 1
|
|
; CHECK-NEXT: [[DOTI1:%.*]] = bitcast float [[A_I1]] to i32
|
|
; CHECK-NEXT: [[IS_FPCLASS_NEGZERO_I0:%.*]] = icmp eq i32 [[DOTI0]], -2147483648
|
|
; CHECK-NEXT: [[IS_FPCLASS_NEGZERO_I1:%.*]] = icmp eq i32 [[DOTI1]], -2147483648
|
|
; CHECK-NEXT: [[IS_FPCLASS_NEGZERO_UPTO0:%.*]] = insertelement <2 x i1> poison, i1 [[IS_FPCLASS_NEGZERO_I0]], i64 0
|
|
; CHECK-NEXT: [[IS_FPCLASS_NEGZERO:%.*]] = insertelement <2 x i1> [[IS_FPCLASS_NEGZERO_UPTO0]], i1 [[IS_FPCLASS_NEGZERO_I1]], i64 1
|
|
; CHECK-NEXT: ret <2 x i1> [[IS_FPCLASS_NEGZERO]]
|
|
;
|
|
entry:
|
|
%0 = call <2 x i1> @llvm.is.fpclass.v2f32(<2 x float> %a, i32 32)
|
|
ret <2 x i1> %0
|
|
}
|
|
|
|
define noundef i1 @isnan(float noundef %a) {
|
|
; CHECK-LABEL: define noundef i1 @isnan(
|
|
; CHECK-SAME: float noundef [[A:%.*]]) {
|
|
; CHECK-NEXT: [[ENTRY:.*:]]
|
|
; CHECK-NEXT: [[TMP0:%.*]] = call i1 @dx.op.isSpecialFloat.f32(i32 8, float [[A]]) #[[ATTR0:[0-9]+]]
|
|
; CHECK-NEXT: ret i1 [[TMP0]]
|
|
;
|
|
entry:
|
|
%0 = call i1 @llvm.is.fpclass.f32(float %a, i32 3)
|
|
ret i1 %0
|
|
}
|
|
|
|
define noundef <2 x i1> @isnanv2(<2 x float> noundef %a) {
|
|
; CHECK-LABEL: define noundef <2 x i1> @isnanv2(
|
|
; CHECK-SAME: <2 x float> noundef [[A:%.*]]) {
|
|
; CHECK-NEXT: [[ENTRY:.*:]]
|
|
; CHECK-NEXT: [[A_I0:%.*]] = extractelement <2 x float> [[A]], i64 0
|
|
; CHECK-NEXT: [[DOTI02:%.*]] = call i1 @dx.op.isSpecialFloat.f32(i32 8, float [[A_I0]]) #[[ATTR0]]
|
|
; CHECK-NEXT: [[A_I1:%.*]] = extractelement <2 x float> [[A]], i64 1
|
|
; CHECK-NEXT: [[DOTI11:%.*]] = call i1 @dx.op.isSpecialFloat.f32(i32 8, float [[A_I1]]) #[[ATTR0]]
|
|
; CHECK-NEXT: [[DOTUPTO0:%.*]] = insertelement <2 x i1> poison, i1 [[DOTI02]], i64 0
|
|
; CHECK-NEXT: [[TMP0:%.*]] = insertelement <2 x i1> [[DOTUPTO0]], i1 [[DOTI11]], i64 1
|
|
; CHECK-NEXT: ret <2 x i1> [[TMP0]]
|
|
;
|
|
entry:
|
|
%0 = call <2 x i1> @llvm.is.fpclass.v2f32(<2 x float> %a, i32 3)
|
|
ret <2 x i1> %0
|
|
}
|
|
|
|
define noundef i1 @isinf(float noundef %a) {
|
|
; CHECK-LABEL: define noundef i1 @isinf(
|
|
; CHECK-SAME: float noundef [[A:%.*]]) {
|
|
; CHECK-NEXT: [[ENTRY:.*:]]
|
|
; CHECK-NEXT: [[TMP0:%.*]] = call i1 @dx.op.isSpecialFloat.f32(i32 9, float [[A]]) #[[ATTR0]]
|
|
; CHECK-NEXT: ret i1 [[TMP0]]
|
|
;
|
|
entry:
|
|
%0 = call i1 @llvm.is.fpclass.f32(float %a, i32 516)
|
|
ret i1 %0
|
|
}
|
|
|
|
define noundef <2 x i1> @isinfv2(<2 x float> noundef %a) {
|
|
; CHECK-LABEL: define noundef <2 x i1> @isinfv2(
|
|
; CHECK-SAME: <2 x float> noundef [[A:%.*]]) {
|
|
; CHECK-NEXT: [[ENTRY:.*:]]
|
|
; CHECK-NEXT: [[A_I0:%.*]] = extractelement <2 x float> [[A]], i64 0
|
|
; CHECK-NEXT: [[DOTI02:%.*]] = call i1 @dx.op.isSpecialFloat.f32(i32 9, float [[A_I0]]) #[[ATTR0]]
|
|
; CHECK-NEXT: [[A_I1:%.*]] = extractelement <2 x float> [[A]], i64 1
|
|
; CHECK-NEXT: [[DOTI11:%.*]] = call i1 @dx.op.isSpecialFloat.f32(i32 9, float [[A_I1]]) #[[ATTR0]]
|
|
; CHECK-NEXT: [[DOTUPTO0:%.*]] = insertelement <2 x i1> poison, i1 [[DOTI02]], i64 0
|
|
; CHECK-NEXT: [[TMP0:%.*]] = insertelement <2 x i1> [[DOTUPTO0]], i1 [[DOTI11]], i64 1
|
|
; CHECK-NEXT: ret <2 x i1> [[TMP0]]
|
|
;
|
|
entry:
|
|
%0 = call <2 x i1> @llvm.is.fpclass.v2f32(<2 x float> %a, i32 516)
|
|
ret <2 x i1> %0
|
|
}
|
|
|
|
define noundef i1 @isfinite(float noundef %a) {
|
|
; CHECK-LABEL: define noundef i1 @isfinite(
|
|
; CHECK-SAME: float noundef [[A:%.*]]) {
|
|
; CHECK-NEXT: [[ENTRY:.*:]]
|
|
; CHECK-NEXT: [[TMP0:%.*]] = call i1 @dx.op.isSpecialFloat.f32(i32 10, float [[A]]) #[[ATTR0]]
|
|
; CHECK-NEXT: ret i1 [[TMP0]]
|
|
;
|
|
entry:
|
|
%0 = call i1 @llvm.is.fpclass.f32(float %a, i32 504)
|
|
ret i1 %0
|
|
}
|
|
|
|
define noundef <2 x i1> @isfinitev2(<2 x float> noundef %a) {
|
|
; CHECK-LABEL: define noundef <2 x i1> @isfinitev2(
|
|
; CHECK-SAME: <2 x float> noundef [[A:%.*]]) {
|
|
; CHECK-NEXT: [[ENTRY:.*:]]
|
|
; CHECK-NEXT: [[A_I0:%.*]] = extractelement <2 x float> [[A]], i64 0
|
|
; CHECK-NEXT: [[DOTI02:%.*]] = call i1 @dx.op.isSpecialFloat.f32(i32 10, float [[A_I0]]) #[[ATTR0]]
|
|
; CHECK-NEXT: [[A_I1:%.*]] = extractelement <2 x float> [[A]], i64 1
|
|
; CHECK-NEXT: [[DOTI11:%.*]] = call i1 @dx.op.isSpecialFloat.f32(i32 10, float [[A_I1]]) #[[ATTR0]]
|
|
; CHECK-NEXT: [[DOTUPTO0:%.*]] = insertelement <2 x i1> poison, i1 [[DOTI02]], i64 0
|
|
; CHECK-NEXT: [[TMP0:%.*]] = insertelement <2 x i1> [[DOTUPTO0]], i1 [[DOTI11]], i64 1
|
|
; CHECK-NEXT: ret <2 x i1> [[TMP0]]
|
|
;
|
|
entry:
|
|
%0 = call <2 x i1> @llvm.is.fpclass.v2f32(<2 x float> %a, i32 504)
|
|
ret <2 x i1> %0
|
|
}
|
|
|
|
define noundef i1 @isnormal(float noundef %a) {
|
|
; CHECK-LABEL: define noundef i1 @isnormal(
|
|
; CHECK-SAME: float noundef [[A:%.*]]) {
|
|
; CHECK-NEXT: [[ENTRY:.*:]]
|
|
; CHECK-NEXT: [[TMP0:%.*]] = call i1 @dx.op.isSpecialFloat.f32(i32 11, float [[A]]) #[[ATTR0]]
|
|
; CHECK-NEXT: ret i1 [[TMP0]]
|
|
;
|
|
entry:
|
|
%0 = call i1 @llvm.is.fpclass.f32(float %a, i32 264)
|
|
ret i1 %0
|
|
}
|
|
|
|
define noundef <2 x i1> @isnormalv2(<2 x float> noundef %a) {
|
|
; CHECK-LABEL: define noundef <2 x i1> @isnormalv2(
|
|
; CHECK-SAME: <2 x float> noundef [[A:%.*]]) {
|
|
; CHECK-NEXT: [[ENTRY:.*:]]
|
|
; CHECK-NEXT: [[A_I0:%.*]] = extractelement <2 x float> [[A]], i64 0
|
|
; CHECK-NEXT: [[DOTI02:%.*]] = call i1 @dx.op.isSpecialFloat.f32(i32 11, float [[A_I0]]) #[[ATTR0]]
|
|
; CHECK-NEXT: [[A_I1:%.*]] = extractelement <2 x float> [[A]], i64 1
|
|
; CHECK-NEXT: [[DOTI11:%.*]] = call i1 @dx.op.isSpecialFloat.f32(i32 11, float [[A_I1]]) #[[ATTR0]]
|
|
; CHECK-NEXT: [[DOTUPTO0:%.*]] = insertelement <2 x i1> poison, i1 [[DOTI02]], i64 0
|
|
; CHECK-NEXT: [[TMP0:%.*]] = insertelement <2 x i1> [[DOTUPTO0]], i1 [[DOTI11]], i64 1
|
|
; CHECK-NEXT: ret <2 x i1> [[TMP0]]
|
|
;
|
|
entry:
|
|
%0 = call <2 x i1> @llvm.is.fpclass.v2f32(<2 x float> %a, i32 264)
|
|
ret <2 x i1> %0
|
|
}
|
|
|
|
declare i1 @llvm.is.fpclass.f32(float, i32 immarg)
|
|
declare <2 x i1> @llvm.is.fpclass.v2f32(<2 x float>, i32 immarg)
|