[llvm] [SelectionDAG] [NFC] Add pre-commit test for PR66701. (PR #66796)
via llvm-commits
llvm-commits at lists.llvm.org
Tue Sep 19 11:21:59 PDT 2023
llvmbot wrote:
<!--LLVM PR SUMMARY COMMENT-->
@llvm/pr-subscribers-backend-amdgpu
<details>
<summary>Changes</summary>
[SelectionDAG] [NFC] Add pre-commit test for PR66701.
---
Full diff: https://github.com/llvm/llvm-project/pull/66796.diff
1 Files Affected:
- (modified) llvm/test/CodeGen/AMDGPU/fma.ll (+24)
``````````diff
diff --git a/llvm/test/CodeGen/AMDGPU/fma.ll b/llvm/test/CodeGen/AMDGPU/fma.ll
index b1db04a7fd8863a..0f8560c1d7628a5 100644
--- a/llvm/test/CodeGen/AMDGPU/fma.ll
+++ b/llvm/test/CodeGen/AMDGPU/fma.ll
@@ -154,3 +154,27 @@ define float @fold_fmul_distributive(float %x, float %y) {
%fmul = fmul contract float %fadd, %x
ret float %fmul
}
+
+; test to make sure contract is not dropped such that we can generate fma from following mul/add
+define amdgpu_kernel void @vec_mul_scalar_add_fma(<2 x float> %a, <2 x float> %b, float %c1, ptr addrspace(1) %inptr) {
+; GFX906-LABEL: vec_mul_scalar_add_fma:
+; GFX906: ; %bb.0:
+; GFX906-NEXT: s_load_dwordx4 s[4:7], s[0:1], 0x24
+; GFX906-NEXT: s_waitcnt lgkmcnt(0)
+; GFX906-NEXT: s_load_dword s5, s[0:1], 0x34
+; GFX906-NEXT: s_load_dwordx2 s[2:3], s[0:1], 0x3c
+; GFX906-NEXT: v_mov_b32_e32 v0, 0
+; GFX906-NEXT: v_mov_b32_e32 v1, s6
+; GFX906-NEXT: v_mul_f32_e32 v1, s4, v1
+; GFX906-NEXT: s_waitcnt lgkmcnt(0)
+; GFX906-NEXT: v_add_f32_e32 v1, s5, v1
+; GFX906-NEXT: global_store_dword v0, v1, s[2:3] offset:4
+; GFX906-NEXT: s_endpgm
+ %gep = getelementptr float, ptr addrspace(1) %inptr, i32 1
+ %c = shufflevector <2 x float> %a, <2 x float> poison, <2 x i32> zeroinitializer
+ %mul = fmul contract <2 x float> %c, %b
+ %elv = extractelement <2 x float> %mul, i64 0
+ %add = fadd contract float %elv, %c1
+ store float %add, ptr addrspace(1) %gep, align 4
+ ret void
+}
``````````
</details>
https://github.com/llvm/llvm-project/pull/66796
More information about the llvm-commits
mailing list