1 ; RUN: llc -march=amdgcn -verify-machineinstrs < %s | FileCheck -check-prefix=GCN -check-prefix=SI %s 2 ; XUN: llc -march=amdgcn -mcpu=tonga -mattr=-flat-for-global -verify-machineinstrs < %s | FileCheck -check-prefix=GCN -check-prefix=VI %s 3 4 ; FIXME: None of these trigger madmk emission anymore. It is still 5 ; possible, but requires the correct registers to be used which is 6 ; hard to trigger. 7 8 declare i32 @llvm.amdgcn.workitem.id.x() nounwind readnone 9 declare float @llvm.fabs.f32(float) nounwind readnone 10 11 ; GCN-LABEL: {{^}}madmk_f32: 12 ; GCN-DAG: buffer_load_dword [[VA:v[0-9]+]], {{v\[[0-9]+:[0-9]+\]}}, {{s\[[0-9]+:[0-9]+\]}}, 0 addr64{{$}} 13 ; GCN-DAG: buffer_load_dword [[VB:v[0-9]+]], {{v\[[0-9]+:[0-9]+\]}}, {{s\[[0-9]+:[0-9]+\]}}, 0 addr64 offset:4 14 ; GCN: v_mac_f32_e32 [[VB]], 0x41200000, [[VA]] 15 define amdgpu_kernel void @madmk_f32(float addrspace(1)* noalias %out, float addrspace(1)* noalias %in) nounwind { 16 %tid = tail call i32 @llvm.amdgcn.workitem.id.x() nounwind readnone 17 %gep.0 = getelementptr float, float addrspace(1)* %in, i32 %tid 18 %gep.1 = getelementptr float, float addrspace(1)* %gep.0, i32 1 19 %out.gep = getelementptr float, float addrspace(1)* %out, i32 %tid 20 21 %a = load volatile float, float addrspace(1)* %gep.0, align 4 22 %b = load volatile float, float addrspace(1)* %gep.1, align 4 23 24 %mul = fmul float %a, 10.0 25 %madmk = fadd float %mul, %b 26 store float %madmk, float addrspace(1)* %out.gep, align 4 27 ret void 28 } 29 30 ; GCN-LABEL: {{^}}madmk_2_use_f32: 31 ; GCN-DAG: buffer_load_dword [[VA:v[0-9]+]], {{v\[[0-9]+:[0-9]+\]}}, {{s\[[0-9]+:[0-9]+\]}}, 0 addr64{{$}} 32 ; GCN-DAG: buffer_load_dword [[VB:v[0-9]+]], {{v\[[0-9]+:[0-9]+\]}}, {{s\[[0-9]+:[0-9]+\]}}, 0 addr64 offset:4 33 ; GCN-DAG: buffer_load_dword [[VC:v[0-9]+]], {{v\[[0-9]+:[0-9]+\]}}, {{s\[[0-9]+:[0-9]+\]}}, 0 addr64 offset:8 34 ; GCN-DAG: v_mov_b32_e32 [[VK:v[0-9]+]], 0x41200000 35 ; GCN-DAG: v_mac_f32_e32 [[VB]], [[VA]], [[VK]] 36 ; GCN-DAG: v_mac_f32_e32 [[VC]], [[VA]], [[VK]] 37 ; GCN: s_endpgm 38 define amdgpu_kernel void @madmk_2_use_f32(float addrspace(1)* noalias %out, float addrspace(1)* noalias %in) nounwind { 39 %tid = tail call i32 @llvm.amdgcn.workitem.id.x() nounwind readnone 40 41 %in.gep.0 = getelementptr float, float addrspace(1)* %in, i32 %tid 42 %in.gep.1 = getelementptr float, float addrspace(1)* %in.gep.0, i32 1 43 %in.gep.2 = getelementptr float, float addrspace(1)* %in.gep.0, i32 2 44 45 %out.gep.0 = getelementptr float, float addrspace(1)* %out, i32 %tid 46 %out.gep.1 = getelementptr float, float addrspace(1)* %in.gep.0, i32 1 47 48 %a = load volatile float, float addrspace(1)* %in.gep.0, align 4 49 %b = load volatile float, float addrspace(1)* %in.gep.1, align 4 50 %c = load volatile float, float addrspace(1)* %in.gep.2, align 4 51 52 %mul0 = fmul float %a, 10.0 53 %mul1 = fmul float %a, 10.0 54 %madmk0 = fadd float %mul0, %b 55 %madmk1 = fadd float %mul1, %c 56 57 store float %madmk0, float addrspace(1)* %out.gep.0, align 4 58 store float %madmk1, float addrspace(1)* %out.gep.1, align 4 59 ret void 60 } 61 62 ; We don't get any benefit if the constant is an inline immediate. 63 ; GCN-LABEL: {{^}}madmk_inline_imm_f32: 64 ; GCN-DAG: buffer_load_dword [[VA:v[0-9]+]], {{v\[[0-9]+:[0-9]+\]}}, {{s\[[0-9]+:[0-9]+\]}}, 0 addr64{{$}} 65 ; GCN-DAG: buffer_load_dword [[VB:v[0-9]+]], {{v\[[0-9]+:[0-9]+\]}}, {{s\[[0-9]+:[0-9]+\]}}, 0 addr64 offset:4 66 ; GCN: v_mac_f32_e32 [[VB]], 4.0, [[VA]] 67 define amdgpu_kernel void @madmk_inline_imm_f32(float addrspace(1)* noalias %out, float addrspace(1)* noalias %in) nounwind { 68 %tid = tail call i32 @llvm.amdgcn.workitem.id.x() nounwind readnone 69 %gep.0 = getelementptr float, float addrspace(1)* %in, i32 %tid 70 %gep.1 = getelementptr float, float addrspace(1)* %gep.0, i32 1 71 %out.gep = getelementptr float, float addrspace(1)* %out, i32 %tid 72 73 %a = load volatile float, float addrspace(1)* %gep.0, align 4 74 %b = load volatile float, float addrspace(1)* %gep.1, align 4 75 76 %mul = fmul float %a, 4.0 77 %madmk = fadd float %mul, %b 78 store float %madmk, float addrspace(1)* %out.gep, align 4 79 ret void 80 } 81 82 ; GCN-LABEL: {{^}}s_s_madmk_f32: 83 ; GCN-NOT: v_madmk_f32 84 ; GCN: v_mac_f32_e32 85 ; GCN: s_endpgm 86 define amdgpu_kernel void @s_s_madmk_f32(float addrspace(1)* noalias %out, [8 x i32], float %a, [8 x i32], float %b) nounwind { 87 %tid = tail call i32 @llvm.amdgcn.workitem.id.x() nounwind readnone 88 %out.gep = getelementptr float, float addrspace(1)* %out, i32 %tid 89 90 %mul = fmul float %a, 10.0 91 %madmk = fadd float %mul, %b 92 store float %madmk, float addrspace(1)* %out.gep, align 4 93 ret void 94 } 95 96 ; GCN-LABEL: {{^}}v_s_madmk_f32: 97 ; GCN-NOT: v_madmk_f32 98 ; GCN: v_mad_f32 99 ; GCN: s_endpgm 100 define amdgpu_kernel void @v_s_madmk_f32(float addrspace(1)* noalias %out, float addrspace(1)* noalias %in, float %b) nounwind { 101 %tid = tail call i32 @llvm.amdgcn.workitem.id.x() nounwind readnone 102 %gep.0 = getelementptr float, float addrspace(1)* %in, i32 %tid 103 %out.gep = getelementptr float, float addrspace(1)* %out, i32 %tid 104 %a = load float, float addrspace(1)* %gep.0, align 4 105 106 %mul = fmul float %a, 10.0 107 %madmk = fadd float %mul, %b 108 store float %madmk, float addrspace(1)* %out.gep, align 4 109 ret void 110 } 111 112 ; GCN-LABEL: {{^}}scalar_vector_madmk_f32: 113 ; GCN-NOT: v_madmk_f32 114 ; GCN: v_mac_f32_e32 115 ; GCN: s_endpgm 116 define amdgpu_kernel void @scalar_vector_madmk_f32(float addrspace(1)* noalias %out, float addrspace(1)* noalias %in, float %a) nounwind { 117 %tid = tail call i32 @llvm.amdgcn.workitem.id.x() nounwind readnone 118 %gep.0 = getelementptr float, float addrspace(1)* %in, i32 %tid 119 %out.gep = getelementptr float, float addrspace(1)* %out, i32 %tid 120 %b = load float, float addrspace(1)* %gep.0, align 4 121 122 %mul = fmul float %a, 10.0 123 %madmk = fadd float %mul, %b 124 store float %madmk, float addrspace(1)* %out.gep, align 4 125 ret void 126 } 127 128 ; GCN-LABEL: {{^}}no_madmk_src0_modifier_f32: 129 ; GCN-DAG: buffer_load_dword [[VA:v[0-9]+]], {{v\[[0-9]+:[0-9]+\]}}, {{s\[[0-9]+:[0-9]+\]}}, 0 addr64{{$}} 130 ; GCN-DAG: buffer_load_dword [[VB:v[0-9]+]], {{v\[[0-9]+:[0-9]+\]}}, {{s\[[0-9]+:[0-9]+\]}}, 0 addr64 offset:4 131 ; GCN-DAG: v_mov_b32_e32 [[VK:v[0-9]+]], 0x41200000 132 ; GCN: v_mad_f32 {{v[0-9]+}}, |[[VA]]|, [[VK]], [[VB]] 133 define amdgpu_kernel void @no_madmk_src0_modifier_f32(float addrspace(1)* noalias %out, float addrspace(1)* noalias %in) nounwind { 134 %tid = tail call i32 @llvm.amdgcn.workitem.id.x() nounwind readnone 135 %gep.0 = getelementptr float, float addrspace(1)* %in, i32 %tid 136 %gep.1 = getelementptr float, float addrspace(1)* %gep.0, i32 1 137 %out.gep = getelementptr float, float addrspace(1)* %out, i32 %tid 138 139 %a = load volatile float, float addrspace(1)* %gep.0, align 4 140 %b = load volatile float, float addrspace(1)* %gep.1, align 4 141 142 %a.fabs = call float @llvm.fabs.f32(float %a) nounwind readnone 143 144 %mul = fmul float %a.fabs, 10.0 145 %madmk = fadd float %mul, %b 146 store float %madmk, float addrspace(1)* %out.gep, align 4 147 ret void 148 } 149 150 ; GCN-LABEL: {{^}}no_madmk_src2_modifier_f32: 151 ; GCN-DAG: buffer_load_dword [[VA:v[0-9]+]], {{v\[[0-9]+:[0-9]+\]}}, {{s\[[0-9]+:[0-9]+\]}}, 0 addr64{{$}} 152 ; GCN-DAG: buffer_load_dword [[VB:v[0-9]+]], {{v\[[0-9]+:[0-9]+\]}}, {{s\[[0-9]+:[0-9]+\]}}, 0 addr64 offset:4 153 ; GCN: v_mad_f32 {{v[0-9]+}}, {{v[0-9]+}}, {{v[0-9]+}}, |{{[sv][0-9]+}}| 154 define amdgpu_kernel void @no_madmk_src2_modifier_f32(float addrspace(1)* noalias %out, float addrspace(1)* noalias %in) nounwind { 155 %tid = tail call i32 @llvm.amdgcn.workitem.id.x() nounwind readnone 156 %gep.0 = getelementptr float, float addrspace(1)* %in, i32 %tid 157 %gep.1 = getelementptr float, float addrspace(1)* %gep.0, i32 1 158 %out.gep = getelementptr float, float addrspace(1)* %out, i32 %tid 159 160 %a = load volatile float, float addrspace(1)* %gep.0, align 4 161 %b = load volatile float, float addrspace(1)* %gep.1, align 4 162 163 %b.fabs = call float @llvm.fabs.f32(float %b) nounwind readnone 164 165 %mul = fmul float %a, 10.0 166 %madmk = fadd float %mul, %b.fabs 167 store float %madmk, float addrspace(1)* %out.gep, align 4 168 ret void 169 } 170 171 ; GCN-LABEL: {{^}}madmk_add_inline_imm_f32: 172 ; GCN: buffer_load_dword [[A:v[0-9]+]] 173 ; GCN: v_mov_b32_e32 [[VK:v[0-9]+]], 0x41200000 174 ; GCN: v_mad_f32 {{v[0-9]+}}, [[A]], [[VK]], 2.0 175 define amdgpu_kernel void @madmk_add_inline_imm_f32(float addrspace(1)* noalias %out, float addrspace(1)* noalias %in) nounwind { 176 %tid = tail call i32 @llvm.amdgcn.workitem.id.x() nounwind readnone 177 %gep.0 = getelementptr float, float addrspace(1)* %in, i32 %tid 178 %out.gep = getelementptr float, float addrspace(1)* %out, i32 %tid 179 180 %a = load float, float addrspace(1)* %gep.0, align 4 181 182 %mul = fmul float %a, 10.0 183 %madmk = fadd float %mul, 2.0 184 store float %madmk, float addrspace(1)* %out.gep, align 4 185 ret void 186 } 187 188 ; SI-LABEL: {{^}}kill_madmk_verifier_error: 189 ; SI: s_xor_b64 190 ; SI: v_mac_f32_e32 {{v[0-9]+}}, 0x472aee8c, {{v[0-9]+}} 191 ; SI: s_or_b64 192 define amdgpu_kernel void @kill_madmk_verifier_error() nounwind { 193 bb: 194 br label %bb2 195 196 bb1: ; preds = %bb2 197 ret void 198 199 bb2: ; preds = %bb6, %bb 200 %tmp = phi float [ undef, %bb ], [ %tmp8, %bb6 ] 201 %tid = call i32 @llvm.amdgcn.mbcnt.lo(i32 -1, i32 0) #1 202 %f_tid = bitcast i32 %tid to float 203 %tmp3 = fsub float %f_tid, %tmp 204 %tmp5 = fcmp oeq float %tmp3, 1.000000e+04 205 br i1 %tmp5, label %bb1, label %bb6 206 207 bb6: ; preds = %bb2 208 %tmp7 = fmul float %tmp, 0x40E55DD180000000 209 %tmp8 = fadd float %tmp7, %tmp 210 br label %bb2 211 } 212 213 declare i32 @llvm.amdgcn.mbcnt.lo(i32, i32) #1 214 215 attributes #1 = { nounwind readnone } 216