Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2014 Advanced Micro Devices, Inc. |
| 3 | * |
| 4 | * Permission is hereby granted, free of charge, to any person obtaining a |
| 5 | * copy of this software and associated documentation files (the |
| 6 | * "Software"), to deal in the Software without restriction, including |
| 7 | * without limitation the rights to use, copy, modify, merge, publish, |
| 8 | * distribute, sub license, and/or sell copies of the Software, and to |
| 9 | * permit persons to whom the Software is furnished to do so, subject to |
| 10 | * the following conditions: |
| 11 | * |
| 12 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| 13 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| 14 | * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL |
| 15 | * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM, |
| 16 | * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR |
| 17 | * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE |
| 18 | * USE OR OTHER DEALINGS IN THE SOFTWARE. |
| 19 | * |
| 20 | * The above copyright notice and this permission notice (including the |
| 21 | * next paragraph) shall be included in all copies or substantial portions |
| 22 | * of the Software. |
| 23 | * |
| 24 | */ |
| 25 | /* based on pieces from si_pipe.c and radeon_llvm_emit.c */ |
| 26 | #include "ac_llvm_build.h" |
| 27 | |
| 28 | #include <llvm-c/Core.h> |
| 29 | |
| 30 | #include "c11/threads.h" |
| 31 | |
| 32 | #include <assert.h> |
| 33 | #include <stdio.h> |
| 34 | |
| 35 | #include "ac_llvm_util.h" |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 36 | #include "ac_exp_param.h" |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 37 | #include "util/bitscan.h" |
| 38 | #include "util/macros.h" |
Connor Abbott | ac27fa7 | 2017-06-05 14:16:43 -0700 | [diff] [blame] | 39 | #include "util/u_atomic.h" |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 40 | #include "sid.h" |
| 41 | |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 42 | #include "shader_enums.h" |
| 43 | |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 44 | /* Initialize module-independent parts of the context. |
| 45 | * |
| 46 | * The caller is responsible for initializing ctx::module and ctx::builder. |
| 47 | */ |
| 48 | void |
| 49 | ac_llvm_context_init(struct ac_llvm_context *ctx, LLVMContextRef context) |
| 50 | { |
| 51 | LLVMValueRef args[1]; |
| 52 | |
| 53 | ctx->context = context; |
| 54 | ctx->module = NULL; |
| 55 | ctx->builder = NULL; |
| 56 | |
| 57 | ctx->voidt = LLVMVoidTypeInContext(ctx->context); |
| 58 | ctx->i1 = LLVMInt1TypeInContext(ctx->context); |
| 59 | ctx->i8 = LLVMInt8TypeInContext(ctx->context); |
Nicolai Hähnle | 7bf8c94 | 2017-03-30 14:10:26 +0200 | [diff] [blame] | 60 | ctx->i16 = LLVMIntTypeInContext(ctx->context, 16); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 61 | ctx->i32 = LLVMIntTypeInContext(ctx->context, 32); |
Nicolai Hähnle | 7bf8c94 | 2017-03-30 14:10:26 +0200 | [diff] [blame] | 62 | ctx->i64 = LLVMIntTypeInContext(ctx->context, 64); |
| 63 | ctx->f16 = LLVMHalfTypeInContext(ctx->context); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 64 | ctx->f32 = LLVMFloatTypeInContext(ctx->context); |
Nicolai Hähnle | 7bf8c94 | 2017-03-30 14:10:26 +0200 | [diff] [blame] | 65 | ctx->f64 = LLVMDoubleTypeInContext(ctx->context); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 66 | ctx->v4i32 = LLVMVectorType(ctx->i32, 4); |
| 67 | ctx->v4f32 = LLVMVectorType(ctx->f32, 4); |
Nicolai Hähnle | edfd3be | 2017-06-08 20:04:28 +0200 | [diff] [blame] | 68 | ctx->v8i32 = LLVMVectorType(ctx->i32, 8); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 69 | |
Nicolai Hähnle | 331a574 | 2017-05-18 22:02:48 +0200 | [diff] [blame] | 70 | ctx->i32_0 = LLVMConstInt(ctx->i32, 0, false); |
| 71 | ctx->i32_1 = LLVMConstInt(ctx->i32, 1, false); |
| 72 | ctx->f32_0 = LLVMConstReal(ctx->f32, 0.0); |
| 73 | ctx->f32_1 = LLVMConstReal(ctx->f32, 1.0); |
| 74 | |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 75 | ctx->range_md_kind = LLVMGetMDKindIDInContext(ctx->context, |
| 76 | "range", 5); |
| 77 | |
| 78 | ctx->invariant_load_md_kind = LLVMGetMDKindIDInContext(ctx->context, |
| 79 | "invariant.load", 14); |
| 80 | |
| 81 | ctx->fpmath_md_kind = LLVMGetMDKindIDInContext(ctx->context, "fpmath", 6); |
| 82 | |
| 83 | args[0] = LLVMConstReal(ctx->f32, 2.5); |
| 84 | ctx->fpmath_md_2p5_ulp = LLVMMDNodeInContext(ctx->context, args, 1); |
| 85 | |
| 86 | ctx->uniform_md_kind = LLVMGetMDKindIDInContext(ctx->context, |
| 87 | "amdgpu.uniform", 14); |
| 88 | |
| 89 | ctx->empty_md = LLVMMDNodeInContext(ctx->context, NULL, 0); |
| 90 | } |
| 91 | |
Connor Abbott | c181d4f | 2017-06-05 14:37:01 -0700 | [diff] [blame] | 92 | unsigned |
| 93 | ac_get_type_size(LLVMTypeRef type) |
| 94 | { |
| 95 | LLVMTypeKind kind = LLVMGetTypeKind(type); |
| 96 | |
| 97 | switch (kind) { |
| 98 | case LLVMIntegerTypeKind: |
| 99 | return LLVMGetIntTypeWidth(type) / 8; |
| 100 | case LLVMFloatTypeKind: |
| 101 | return 4; |
| 102 | case LLVMPointerTypeKind: |
| 103 | return 8; |
| 104 | case LLVMVectorTypeKind: |
| 105 | return LLVMGetVectorSize(type) * |
| 106 | ac_get_type_size(LLVMGetElementType(type)); |
| 107 | case LLVMArrayTypeKind: |
| 108 | return LLVMGetArrayLength(type) * |
| 109 | ac_get_type_size(LLVMGetElementType(type)); |
| 110 | default: |
| 111 | assert(0); |
| 112 | return 0; |
| 113 | } |
| 114 | } |
| 115 | |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 116 | LLVMValueRef |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 117 | ac_build_intrinsic(struct ac_llvm_context *ctx, const char *name, |
| 118 | LLVMTypeRef return_type, LLVMValueRef *params, |
| 119 | unsigned param_count, unsigned attrib_mask) |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 120 | { |
Marek Olšák | 940da36 | 2017-02-22 02:29:12 +0100 | [diff] [blame] | 121 | LLVMValueRef function, call; |
| 122 | bool set_callsite_attrs = HAVE_LLVM >= 0x0400 && |
| 123 | !(attrib_mask & AC_FUNC_ATTR_LEGACY); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 124 | |
| 125 | function = LLVMGetNamedFunction(ctx->module, name); |
| 126 | if (!function) { |
| 127 | LLVMTypeRef param_types[32], function_type; |
| 128 | unsigned i; |
| 129 | |
| 130 | assert(param_count <= 32); |
| 131 | |
| 132 | for (i = 0; i < param_count; ++i) { |
| 133 | assert(params[i]); |
| 134 | param_types[i] = LLVMTypeOf(params[i]); |
| 135 | } |
| 136 | function_type = |
| 137 | LLVMFunctionType(return_type, param_types, param_count, 0); |
| 138 | function = LLVMAddFunction(ctx->module, name, function_type); |
| 139 | |
| 140 | LLVMSetFunctionCallConv(function, LLVMCCallConv); |
| 141 | LLVMSetLinkage(function, LLVMExternalLinkage); |
| 142 | |
Marek Olšák | 940da36 | 2017-02-22 02:29:12 +0100 | [diff] [blame] | 143 | if (!set_callsite_attrs) |
| 144 | ac_add_func_attributes(ctx->context, function, attrib_mask); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 145 | } |
Marek Olšák | 940da36 | 2017-02-22 02:29:12 +0100 | [diff] [blame] | 146 | |
| 147 | call = LLVMBuildCall(ctx->builder, function, params, param_count, ""); |
| 148 | if (set_callsite_attrs) |
| 149 | ac_add_func_attributes(ctx->context, call, attrib_mask); |
| 150 | return call; |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 151 | } |
| 152 | |
Marek Olšák | 4b2e5b9 | 2017-02-23 23:37:59 +0100 | [diff] [blame] | 153 | static LLVMValueRef bitcast_to_float(struct ac_llvm_context *ctx, |
| 154 | LLVMValueRef value) |
| 155 | { |
| 156 | LLVMTypeRef type = LLVMTypeOf(value); |
| 157 | LLVMTypeRef new_type; |
| 158 | |
| 159 | if (LLVMGetTypeKind(type) == LLVMVectorTypeKind) |
| 160 | new_type = LLVMVectorType(ctx->f32, LLVMGetVectorSize(type)); |
| 161 | else |
| 162 | new_type = ctx->f32; |
| 163 | |
| 164 | return LLVMBuildBitCast(ctx->builder, value, new_type, ""); |
| 165 | } |
| 166 | |
Marek Olšák | 9af0331 | 2017-02-23 22:58:49 +0100 | [diff] [blame] | 167 | /** |
| 168 | * Given the i32 or vNi32 \p type, generate the textual name (e.g. for use with |
| 169 | * intrinsic names). |
| 170 | */ |
| 171 | void ac_build_type_name_for_intr(LLVMTypeRef type, char *buf, unsigned bufsize) |
| 172 | { |
| 173 | LLVMTypeRef elem_type = type; |
| 174 | |
| 175 | assert(bufsize >= 8); |
| 176 | |
| 177 | if (LLVMGetTypeKind(type) == LLVMVectorTypeKind) { |
| 178 | int ret = snprintf(buf, bufsize, "v%u", |
| 179 | LLVMGetVectorSize(type)); |
| 180 | if (ret < 0) { |
| 181 | char *type_name = LLVMPrintTypeToString(type); |
| 182 | fprintf(stderr, "Error building type name for: %s\n", |
| 183 | type_name); |
| 184 | return; |
| 185 | } |
| 186 | elem_type = LLVMGetElementType(type); |
| 187 | buf += ret; |
| 188 | bufsize -= ret; |
| 189 | } |
| 190 | switch (LLVMGetTypeKind(elem_type)) { |
| 191 | default: break; |
| 192 | case LLVMIntegerTypeKind: |
| 193 | snprintf(buf, bufsize, "i%d", LLVMGetIntTypeWidth(elem_type)); |
| 194 | break; |
| 195 | case LLVMFloatTypeKind: |
| 196 | snprintf(buf, bufsize, "f32"); |
| 197 | break; |
| 198 | case LLVMDoubleTypeKind: |
| 199 | snprintf(buf, bufsize, "f64"); |
| 200 | break; |
| 201 | } |
| 202 | } |
| 203 | |
Connor Abbott | ac27fa7 | 2017-06-05 14:16:43 -0700 | [diff] [blame] | 204 | /* Prevent optimizations (at least of memory accesses) across the current |
| 205 | * point in the program by emitting empty inline assembly that is marked as |
| 206 | * having side effects. |
| 207 | * |
| 208 | * Optionally, a value can be passed through the inline assembly to prevent |
| 209 | * LLVM from hoisting calls to ReadNone functions. |
| 210 | */ |
| 211 | void |
| 212 | ac_build_optimization_barrier(struct ac_llvm_context *ctx, |
| 213 | LLVMValueRef *pvgpr) |
| 214 | { |
| 215 | static int counter = 0; |
| 216 | |
| 217 | LLVMBuilderRef builder = ctx->builder; |
| 218 | char code[16]; |
| 219 | |
| 220 | snprintf(code, sizeof(code), "; %d", p_atomic_inc_return(&counter)); |
| 221 | |
| 222 | if (!pvgpr) { |
| 223 | LLVMTypeRef ftype = LLVMFunctionType(ctx->voidt, NULL, 0, false); |
| 224 | LLVMValueRef inlineasm = LLVMConstInlineAsm(ftype, code, "", true, false); |
| 225 | LLVMBuildCall(builder, inlineasm, NULL, 0, ""); |
| 226 | } else { |
| 227 | LLVMTypeRef ftype = LLVMFunctionType(ctx->i32, &ctx->i32, 1, false); |
| 228 | LLVMValueRef inlineasm = LLVMConstInlineAsm(ftype, code, "=v,0", true, false); |
| 229 | LLVMValueRef vgpr = *pvgpr; |
| 230 | LLVMTypeRef vgpr_type = LLVMTypeOf(vgpr); |
| 231 | unsigned vgpr_size = ac_get_type_size(vgpr_type); |
| 232 | LLVMValueRef vgpr0; |
| 233 | |
| 234 | assert(vgpr_size % 4 == 0); |
| 235 | |
| 236 | vgpr = LLVMBuildBitCast(builder, vgpr, LLVMVectorType(ctx->i32, vgpr_size / 4), ""); |
| 237 | vgpr0 = LLVMBuildExtractElement(builder, vgpr, ctx->i32_0, ""); |
| 238 | vgpr0 = LLVMBuildCall(builder, inlineasm, &vgpr0, 1, ""); |
| 239 | vgpr = LLVMBuildInsertElement(builder, vgpr, vgpr0, ctx->i32_0, ""); |
| 240 | vgpr = LLVMBuildBitCast(builder, vgpr, vgpr_type, ""); |
| 241 | |
| 242 | *pvgpr = vgpr; |
| 243 | } |
| 244 | } |
| 245 | |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 246 | LLVMValueRef |
Connor Abbott | bd73b89 | 2017-06-05 15:20:04 -0700 | [diff] [blame^] | 247 | ac_build_ballot(struct ac_llvm_context *ctx, |
| 248 | LLVMValueRef value) |
| 249 | { |
| 250 | LLVMValueRef args[3] = { |
| 251 | value, |
| 252 | ctx->i32_0, |
| 253 | LLVMConstInt(ctx->i32, LLVMIntNE, 0) |
| 254 | }; |
| 255 | |
| 256 | /* We currently have no other way to prevent LLVM from lifting the icmp |
| 257 | * calls to a dominating basic block. |
| 258 | */ |
| 259 | ac_build_optimization_barrier(ctx, &args[0]); |
| 260 | |
| 261 | if (LLVMTypeOf(args[0]) != ctx->i32) |
| 262 | args[0] = LLVMBuildBitCast(ctx->builder, args[0], ctx->i32, ""); |
| 263 | |
| 264 | return ac_build_intrinsic(ctx, |
| 265 | "llvm.amdgcn.icmp.i32", |
| 266 | ctx->i64, args, 3, |
| 267 | AC_FUNC_ATTR_NOUNWIND | |
| 268 | AC_FUNC_ATTR_READNONE | |
| 269 | AC_FUNC_ATTR_CONVERGENT); |
| 270 | } |
| 271 | |
| 272 | LLVMValueRef |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 273 | ac_build_gather_values_extended(struct ac_llvm_context *ctx, |
| 274 | LLVMValueRef *values, |
| 275 | unsigned value_count, |
| 276 | unsigned value_stride, |
Nicolai Hähnle | ac2ab5a | 2017-06-25 13:04:51 +0200 | [diff] [blame] | 277 | bool load, |
| 278 | bool always_vector) |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 279 | { |
| 280 | LLVMBuilderRef builder = ctx->builder; |
Marek Olšák | c7878b0 | 2017-02-23 01:34:27 +0100 | [diff] [blame] | 281 | LLVMValueRef vec = NULL; |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 282 | unsigned i; |
| 283 | |
Nicolai Hähnle | ac2ab5a | 2017-06-25 13:04:51 +0200 | [diff] [blame] | 284 | if (value_count == 1 && !always_vector) { |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 285 | if (load) |
| 286 | return LLVMBuildLoad(builder, values[0], ""); |
| 287 | return values[0]; |
| 288 | } else if (!value_count) |
| 289 | unreachable("value_count is 0"); |
| 290 | |
| 291 | for (i = 0; i < value_count; i++) { |
| 292 | LLVMValueRef value = values[i * value_stride]; |
| 293 | if (load) |
| 294 | value = LLVMBuildLoad(builder, value, ""); |
| 295 | |
| 296 | if (!i) |
| 297 | vec = LLVMGetUndef( LLVMVectorType(LLVMTypeOf(value), value_count)); |
| 298 | LLVMValueRef index = LLVMConstInt(ctx->i32, i, false); |
| 299 | vec = LLVMBuildInsertElement(builder, vec, value, index, ""); |
| 300 | } |
| 301 | return vec; |
| 302 | } |
| 303 | |
| 304 | LLVMValueRef |
| 305 | ac_build_gather_values(struct ac_llvm_context *ctx, |
| 306 | LLVMValueRef *values, |
| 307 | unsigned value_count) |
| 308 | { |
Nicolai Hähnle | ac2ab5a | 2017-06-25 13:04:51 +0200 | [diff] [blame] | 309 | return ac_build_gather_values_extended(ctx, values, value_count, 1, false, false); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 310 | } |
| 311 | |
| 312 | LLVMValueRef |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 313 | ac_build_fdiv(struct ac_llvm_context *ctx, |
| 314 | LLVMValueRef num, |
| 315 | LLVMValueRef den) |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 316 | { |
| 317 | LLVMValueRef ret = LLVMBuildFDiv(ctx->builder, num, den, ""); |
| 318 | |
| 319 | if (!LLVMIsConstant(ret)) |
| 320 | LLVMSetMetadata(ret, ctx->fpmath_md_kind, ctx->fpmath_md_2p5_ulp); |
| 321 | return ret; |
| 322 | } |
| 323 | |
| 324 | /* Coordinates for cube map selection. sc, tc, and ma are as in Table 8.27 |
| 325 | * of the OpenGL 4.5 (Compatibility Profile) specification, except ma is |
| 326 | * already multiplied by two. id is the cube face number. |
| 327 | */ |
| 328 | struct cube_selection_coords { |
| 329 | LLVMValueRef stc[2]; |
| 330 | LLVMValueRef ma; |
| 331 | LLVMValueRef id; |
| 332 | }; |
| 333 | |
| 334 | static void |
| 335 | build_cube_intrinsic(struct ac_llvm_context *ctx, |
| 336 | LLVMValueRef in[3], |
| 337 | struct cube_selection_coords *out) |
| 338 | { |
Marek Olšák | 12beef0 | 2017-04-25 02:18:10 +0200 | [diff] [blame] | 339 | LLVMTypeRef f32 = ctx->f32; |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 340 | |
Marek Olšák | 12beef0 | 2017-04-25 02:18:10 +0200 | [diff] [blame] | 341 | out->stc[1] = ac_build_intrinsic(ctx, "llvm.amdgcn.cubetc", |
| 342 | f32, in, 3, AC_FUNC_ATTR_READNONE); |
| 343 | out->stc[0] = ac_build_intrinsic(ctx, "llvm.amdgcn.cubesc", |
| 344 | f32, in, 3, AC_FUNC_ATTR_READNONE); |
| 345 | out->ma = ac_build_intrinsic(ctx, "llvm.amdgcn.cubema", |
| 346 | f32, in, 3, AC_FUNC_ATTR_READNONE); |
| 347 | out->id = ac_build_intrinsic(ctx, "llvm.amdgcn.cubeid", |
| 348 | f32, in, 3, AC_FUNC_ATTR_READNONE); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 349 | } |
| 350 | |
| 351 | /** |
| 352 | * Build a manual selection sequence for cube face sc/tc coordinates and |
| 353 | * major axis vector (multiplied by 2 for consistency) for the given |
| 354 | * vec3 \p coords, for the face implied by \p selcoords. |
| 355 | * |
| 356 | * For the major axis, we always adjust the sign to be in the direction of |
| 357 | * selcoords.ma; i.e., a positive out_ma means that coords is pointed towards |
| 358 | * the selcoords major axis. |
| 359 | */ |
| 360 | static void build_cube_select(LLVMBuilderRef builder, |
| 361 | const struct cube_selection_coords *selcoords, |
| 362 | const LLVMValueRef *coords, |
| 363 | LLVMValueRef *out_st, |
| 364 | LLVMValueRef *out_ma) |
| 365 | { |
| 366 | LLVMTypeRef f32 = LLVMTypeOf(coords[0]); |
| 367 | LLVMValueRef is_ma_positive; |
| 368 | LLVMValueRef sgn_ma; |
| 369 | LLVMValueRef is_ma_z, is_not_ma_z; |
| 370 | LLVMValueRef is_ma_y; |
| 371 | LLVMValueRef is_ma_x; |
| 372 | LLVMValueRef sgn; |
| 373 | LLVMValueRef tmp; |
| 374 | |
| 375 | is_ma_positive = LLVMBuildFCmp(builder, LLVMRealUGE, |
| 376 | selcoords->ma, LLVMConstReal(f32, 0.0), ""); |
| 377 | sgn_ma = LLVMBuildSelect(builder, is_ma_positive, |
| 378 | LLVMConstReal(f32, 1.0), LLVMConstReal(f32, -1.0), ""); |
| 379 | |
| 380 | is_ma_z = LLVMBuildFCmp(builder, LLVMRealUGE, selcoords->id, LLVMConstReal(f32, 4.0), ""); |
| 381 | is_not_ma_z = LLVMBuildNot(builder, is_ma_z, ""); |
| 382 | is_ma_y = LLVMBuildAnd(builder, is_not_ma_z, |
| 383 | LLVMBuildFCmp(builder, LLVMRealUGE, selcoords->id, LLVMConstReal(f32, 2.0), ""), ""); |
| 384 | is_ma_x = LLVMBuildAnd(builder, is_not_ma_z, LLVMBuildNot(builder, is_ma_y, ""), ""); |
| 385 | |
| 386 | /* Select sc */ |
| 387 | tmp = LLVMBuildSelect(builder, is_ma_z, coords[2], coords[0], ""); |
| 388 | sgn = LLVMBuildSelect(builder, is_ma_y, LLVMConstReal(f32, 1.0), |
| 389 | LLVMBuildSelect(builder, is_ma_x, sgn_ma, |
| 390 | LLVMBuildFNeg(builder, sgn_ma, ""), ""), ""); |
| 391 | out_st[0] = LLVMBuildFMul(builder, tmp, sgn, ""); |
| 392 | |
| 393 | /* Select tc */ |
| 394 | tmp = LLVMBuildSelect(builder, is_ma_y, coords[2], coords[1], ""); |
| 395 | sgn = LLVMBuildSelect(builder, is_ma_y, LLVMBuildFNeg(builder, sgn_ma, ""), |
| 396 | LLVMConstReal(f32, -1.0), ""); |
| 397 | out_st[1] = LLVMBuildFMul(builder, tmp, sgn, ""); |
| 398 | |
| 399 | /* Select ma */ |
| 400 | tmp = LLVMBuildSelect(builder, is_ma_z, coords[2], |
| 401 | LLVMBuildSelect(builder, is_ma_y, coords[1], coords[0], ""), ""); |
| 402 | sgn = LLVMBuildSelect(builder, is_ma_positive, |
| 403 | LLVMConstReal(f32, 2.0), LLVMConstReal(f32, -2.0), ""); |
| 404 | *out_ma = LLVMBuildFMul(builder, tmp, sgn, ""); |
| 405 | } |
| 406 | |
| 407 | void |
| 408 | ac_prepare_cube_coords(struct ac_llvm_context *ctx, |
| 409 | bool is_deriv, bool is_array, |
| 410 | LLVMValueRef *coords_arg, |
| 411 | LLVMValueRef *derivs_arg) |
| 412 | { |
| 413 | |
| 414 | LLVMBuilderRef builder = ctx->builder; |
| 415 | struct cube_selection_coords selcoords; |
| 416 | LLVMValueRef coords[3]; |
| 417 | LLVMValueRef invma; |
| 418 | |
| 419 | build_cube_intrinsic(ctx, coords_arg, &selcoords); |
| 420 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 421 | invma = ac_build_intrinsic(ctx, "llvm.fabs.f32", |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 422 | ctx->f32, &selcoords.ma, 1, AC_FUNC_ATTR_READNONE); |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 423 | invma = ac_build_fdiv(ctx, LLVMConstReal(ctx->f32, 1.0), invma); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 424 | |
| 425 | for (int i = 0; i < 2; ++i) |
| 426 | coords[i] = LLVMBuildFMul(builder, selcoords.stc[i], invma, ""); |
| 427 | |
| 428 | coords[2] = selcoords.id; |
| 429 | |
| 430 | if (is_deriv && derivs_arg) { |
| 431 | LLVMValueRef derivs[4]; |
| 432 | int axis; |
| 433 | |
| 434 | /* Convert cube derivatives to 2D derivatives. */ |
| 435 | for (axis = 0; axis < 2; axis++) { |
| 436 | LLVMValueRef deriv_st[2]; |
| 437 | LLVMValueRef deriv_ma; |
| 438 | |
| 439 | /* Transform the derivative alongside the texture |
| 440 | * coordinate. Mathematically, the correct formula is |
| 441 | * as follows. Assume we're projecting onto the +Z face |
| 442 | * and denote by dx/dh the derivative of the (original) |
| 443 | * X texture coordinate with respect to horizontal |
| 444 | * window coordinates. The projection onto the +Z face |
| 445 | * plane is: |
| 446 | * |
| 447 | * f(x,z) = x/z |
| 448 | * |
| 449 | * Then df/dh = df/dx * dx/dh + df/dz * dz/dh |
| 450 | * = 1/z * dx/dh - x/z * 1/z * dz/dh. |
| 451 | * |
| 452 | * This motivatives the implementation below. |
| 453 | * |
| 454 | * Whether this actually gives the expected results for |
| 455 | * apps that might feed in derivatives obtained via |
| 456 | * finite differences is anyone's guess. The OpenGL spec |
| 457 | * seems awfully quiet about how textureGrad for cube |
| 458 | * maps should be handled. |
| 459 | */ |
| 460 | build_cube_select(builder, &selcoords, &derivs_arg[axis * 3], |
| 461 | deriv_st, &deriv_ma); |
| 462 | |
| 463 | deriv_ma = LLVMBuildFMul(builder, deriv_ma, invma, ""); |
| 464 | |
| 465 | for (int i = 0; i < 2; ++i) |
| 466 | derivs[axis * 2 + i] = |
| 467 | LLVMBuildFSub(builder, |
| 468 | LLVMBuildFMul(builder, deriv_st[i], invma, ""), |
| 469 | LLVMBuildFMul(builder, deriv_ma, coords[i], ""), ""); |
| 470 | } |
| 471 | |
| 472 | memcpy(derivs_arg, derivs, sizeof(derivs)); |
| 473 | } |
| 474 | |
| 475 | /* Shift the texture coordinate. This must be applied after the |
| 476 | * derivative calculation. |
| 477 | */ |
| 478 | for (int i = 0; i < 2; ++i) |
| 479 | coords[i] = LLVMBuildFAdd(builder, coords[i], LLVMConstReal(ctx->f32, 1.5), ""); |
| 480 | |
| 481 | if (is_array) { |
| 482 | /* for cube arrays coord.z = coord.w(array_index) * 8 + face */ |
| 483 | /* coords_arg.w component - array_index for cube arrays */ |
| 484 | LLVMValueRef tmp = LLVMBuildFMul(ctx->builder, coords_arg[3], LLVMConstReal(ctx->f32, 8.0), ""); |
| 485 | coords[2] = LLVMBuildFAdd(ctx->builder, tmp, coords[2], ""); |
| 486 | } |
| 487 | |
| 488 | memcpy(coords_arg, coords, sizeof(coords)); |
| 489 | } |
| 490 | |
| 491 | |
| 492 | LLVMValueRef |
| 493 | ac_build_fs_interp(struct ac_llvm_context *ctx, |
| 494 | LLVMValueRef llvm_chan, |
| 495 | LLVMValueRef attr_number, |
| 496 | LLVMValueRef params, |
| 497 | LLVMValueRef i, |
| 498 | LLVMValueRef j) |
| 499 | { |
| 500 | LLVMValueRef args[5]; |
| 501 | LLVMValueRef p1; |
| 502 | |
| 503 | if (HAVE_LLVM < 0x0400) { |
| 504 | LLVMValueRef ij[2]; |
| 505 | ij[0] = LLVMBuildBitCast(ctx->builder, i, ctx->i32, ""); |
| 506 | ij[1] = LLVMBuildBitCast(ctx->builder, j, ctx->i32, ""); |
| 507 | |
| 508 | args[0] = llvm_chan; |
| 509 | args[1] = attr_number; |
| 510 | args[2] = params; |
| 511 | args[3] = ac_build_gather_values(ctx, ij, 2); |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 512 | return ac_build_intrinsic(ctx, "llvm.SI.fs.interp", |
| 513 | ctx->f32, args, 4, |
| 514 | AC_FUNC_ATTR_READNONE); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 515 | } |
| 516 | |
| 517 | args[0] = i; |
| 518 | args[1] = llvm_chan; |
| 519 | args[2] = attr_number; |
| 520 | args[3] = params; |
| 521 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 522 | p1 = ac_build_intrinsic(ctx, "llvm.amdgcn.interp.p1", |
| 523 | ctx->f32, args, 4, AC_FUNC_ATTR_READNONE); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 524 | |
| 525 | args[0] = p1; |
| 526 | args[1] = j; |
| 527 | args[2] = llvm_chan; |
| 528 | args[3] = attr_number; |
| 529 | args[4] = params; |
| 530 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 531 | return ac_build_intrinsic(ctx, "llvm.amdgcn.interp.p2", |
| 532 | ctx->f32, args, 5, AC_FUNC_ATTR_READNONE); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 533 | } |
| 534 | |
| 535 | LLVMValueRef |
| 536 | ac_build_fs_interp_mov(struct ac_llvm_context *ctx, |
| 537 | LLVMValueRef parameter, |
| 538 | LLVMValueRef llvm_chan, |
| 539 | LLVMValueRef attr_number, |
| 540 | LLVMValueRef params) |
| 541 | { |
| 542 | LLVMValueRef args[4]; |
| 543 | if (HAVE_LLVM < 0x0400) { |
| 544 | args[0] = llvm_chan; |
| 545 | args[1] = attr_number; |
| 546 | args[2] = params; |
| 547 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 548 | return ac_build_intrinsic(ctx, |
| 549 | "llvm.SI.fs.constant", |
| 550 | ctx->f32, args, 3, |
| 551 | AC_FUNC_ATTR_READNONE); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 552 | } |
| 553 | |
| 554 | args[0] = parameter; |
| 555 | args[1] = llvm_chan; |
| 556 | args[2] = attr_number; |
| 557 | args[3] = params; |
| 558 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 559 | return ac_build_intrinsic(ctx, "llvm.amdgcn.interp.mov", |
| 560 | ctx->f32, args, 4, AC_FUNC_ATTR_READNONE); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 561 | } |
| 562 | |
| 563 | LLVMValueRef |
| 564 | ac_build_gep0(struct ac_llvm_context *ctx, |
| 565 | LLVMValueRef base_ptr, |
| 566 | LLVMValueRef index) |
| 567 | { |
| 568 | LLVMValueRef indices[2] = { |
| 569 | LLVMConstInt(ctx->i32, 0, 0), |
| 570 | index, |
| 571 | }; |
| 572 | return LLVMBuildGEP(ctx->builder, base_ptr, |
| 573 | indices, 2, ""); |
| 574 | } |
| 575 | |
| 576 | void |
| 577 | ac_build_indexed_store(struct ac_llvm_context *ctx, |
| 578 | LLVMValueRef base_ptr, LLVMValueRef index, |
| 579 | LLVMValueRef value) |
| 580 | { |
| 581 | LLVMBuildStore(ctx->builder, value, |
| 582 | ac_build_gep0(ctx, base_ptr, index)); |
| 583 | } |
| 584 | |
| 585 | /** |
| 586 | * Build an LLVM bytecode indexed load using LLVMBuildGEP + LLVMBuildLoad. |
| 587 | * It's equivalent to doing a load from &base_ptr[index]. |
| 588 | * |
| 589 | * \param base_ptr Where the array starts. |
| 590 | * \param index The element index into the array. |
| 591 | * \param uniform Whether the base_ptr and index can be assumed to be |
| 592 | * dynamically uniform |
| 593 | */ |
| 594 | LLVMValueRef |
| 595 | ac_build_indexed_load(struct ac_llvm_context *ctx, |
| 596 | LLVMValueRef base_ptr, LLVMValueRef index, |
| 597 | bool uniform) |
| 598 | { |
| 599 | LLVMValueRef pointer; |
| 600 | |
| 601 | pointer = ac_build_gep0(ctx, base_ptr, index); |
| 602 | if (uniform) |
| 603 | LLVMSetMetadata(pointer, ctx->uniform_md_kind, ctx->empty_md); |
| 604 | return LLVMBuildLoad(ctx->builder, pointer, ""); |
| 605 | } |
| 606 | |
| 607 | /** |
| 608 | * Do a load from &base_ptr[index], but also add a flag that it's loading |
| 609 | * a constant from a dynamically uniform index. |
| 610 | */ |
| 611 | LLVMValueRef |
| 612 | ac_build_indexed_load_const(struct ac_llvm_context *ctx, |
| 613 | LLVMValueRef base_ptr, LLVMValueRef index) |
| 614 | { |
| 615 | LLVMValueRef result = ac_build_indexed_load(ctx, base_ptr, index, true); |
| 616 | LLVMSetMetadata(result, ctx->invariant_load_md_kind, ctx->empty_md); |
| 617 | return result; |
| 618 | } |
| 619 | |
| 620 | /* TBUFFER_STORE_FORMAT_{X,XY,XYZ,XYZW} <- the suffix is selected by num_channels=1..4. |
| 621 | * The type of vdata must be one of i32 (num_channels=1), v2i32 (num_channels=2), |
| 622 | * or v4i32 (num_channels=3,4). |
| 623 | */ |
| 624 | void |
Marek Olšák | 27439df | 2017-02-24 01:20:35 +0100 | [diff] [blame] | 625 | ac_build_buffer_store_dword(struct ac_llvm_context *ctx, |
| 626 | LLVMValueRef rsrc, |
| 627 | LLVMValueRef vdata, |
| 628 | unsigned num_channels, |
Marek Olšák | 8cfdbba | 2017-02-24 20:23:23 +0100 | [diff] [blame] | 629 | LLVMValueRef voffset, |
Marek Olšák | 27439df | 2017-02-24 01:20:35 +0100 | [diff] [blame] | 630 | LLVMValueRef soffset, |
| 631 | unsigned inst_offset, |
Marek Olšák | 27439df | 2017-02-24 01:20:35 +0100 | [diff] [blame] | 632 | bool glc, |
Marek Olšák | 97e21cf | 2017-02-24 02:09:47 +0100 | [diff] [blame] | 633 | bool slc, |
| 634 | bool writeonly_memory, |
| 635 | bool has_add_tid) |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 636 | { |
Marek Olšák | 97e21cf | 2017-02-24 02:09:47 +0100 | [diff] [blame] | 637 | /* TODO: Fix stores with ADD_TID and remove the "has_add_tid" flag. */ |
Marek Olšák | 12beef0 | 2017-04-25 02:18:10 +0200 | [diff] [blame] | 638 | if (!has_add_tid) { |
Marek Olšák | 97e21cf | 2017-02-24 02:09:47 +0100 | [diff] [blame] | 639 | /* Split 3 channel stores, becase LLVM doesn't support 3-channel |
| 640 | * intrinsics. */ |
| 641 | if (num_channels == 3) { |
| 642 | LLVMValueRef v[3], v01; |
| 643 | |
| 644 | for (int i = 0; i < 3; i++) { |
| 645 | v[i] = LLVMBuildExtractElement(ctx->builder, vdata, |
| 646 | LLVMConstInt(ctx->i32, i, 0), ""); |
| 647 | } |
| 648 | v01 = ac_build_gather_values(ctx, v, 2); |
| 649 | |
| 650 | ac_build_buffer_store_dword(ctx, rsrc, v01, 2, voffset, |
| 651 | soffset, inst_offset, glc, slc, |
| 652 | writeonly_memory, has_add_tid); |
| 653 | ac_build_buffer_store_dword(ctx, rsrc, v[2], 1, voffset, |
| 654 | soffset, inst_offset + 8, |
| 655 | glc, slc, |
| 656 | writeonly_memory, has_add_tid); |
| 657 | return; |
| 658 | } |
| 659 | |
| 660 | unsigned func = CLAMP(num_channels, 1, 3) - 1; |
| 661 | static const char *types[] = {"f32", "v2f32", "v4f32"}; |
| 662 | char name[256]; |
| 663 | LLVMValueRef offset = soffset; |
| 664 | |
| 665 | if (inst_offset) |
| 666 | offset = LLVMBuildAdd(ctx->builder, offset, |
| 667 | LLVMConstInt(ctx->i32, inst_offset, 0), ""); |
| 668 | if (voffset) |
| 669 | offset = LLVMBuildAdd(ctx->builder, offset, voffset, ""); |
| 670 | |
| 671 | LLVMValueRef args[] = { |
| 672 | bitcast_to_float(ctx, vdata), |
| 673 | LLVMBuildBitCast(ctx->builder, rsrc, ctx->v4i32, ""), |
| 674 | LLVMConstInt(ctx->i32, 0, 0), |
| 675 | offset, |
| 676 | LLVMConstInt(ctx->i1, glc, 0), |
| 677 | LLVMConstInt(ctx->i1, slc, 0), |
| 678 | }; |
| 679 | |
| 680 | snprintf(name, sizeof(name), "llvm.amdgcn.buffer.store.%s", |
| 681 | types[func]); |
| 682 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 683 | ac_build_intrinsic(ctx, name, ctx->voidt, |
| 684 | args, ARRAY_SIZE(args), |
| 685 | writeonly_memory ? |
| 686 | AC_FUNC_ATTR_INACCESSIBLE_MEM_ONLY : |
| 687 | AC_FUNC_ATTR_WRITEONLY); |
Marek Olšák | 97e21cf | 2017-02-24 02:09:47 +0100 | [diff] [blame] | 688 | return; |
| 689 | } |
| 690 | |
Marek Olšák | 27439df | 2017-02-24 01:20:35 +0100 | [diff] [blame] | 691 | static unsigned dfmt[] = { |
| 692 | V_008F0C_BUF_DATA_FORMAT_32, |
| 693 | V_008F0C_BUF_DATA_FORMAT_32_32, |
| 694 | V_008F0C_BUF_DATA_FORMAT_32_32_32, |
| 695 | V_008F0C_BUF_DATA_FORMAT_32_32_32_32 |
| 696 | }; |
| 697 | assert(num_channels >= 1 && num_channels <= 4); |
| 698 | |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 699 | LLVMValueRef args[] = { |
| 700 | rsrc, |
| 701 | vdata, |
| 702 | LLVMConstInt(ctx->i32, num_channels, 0), |
Marek Olšák | 8cfdbba | 2017-02-24 20:23:23 +0100 | [diff] [blame] | 703 | voffset ? voffset : LLVMGetUndef(ctx->i32), |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 704 | soffset, |
| 705 | LLVMConstInt(ctx->i32, inst_offset, 0), |
Marek Olšák | 27439df | 2017-02-24 01:20:35 +0100 | [diff] [blame] | 706 | LLVMConstInt(ctx->i32, dfmt[num_channels - 1], 0), |
| 707 | LLVMConstInt(ctx->i32, V_008F0C_BUF_NUM_FORMAT_UINT, 0), |
Marek Olšák | 8cfdbba | 2017-02-24 20:23:23 +0100 | [diff] [blame] | 708 | LLVMConstInt(ctx->i32, voffset != NULL, 0), |
Marek Olšák | 27439df | 2017-02-24 01:20:35 +0100 | [diff] [blame] | 709 | LLVMConstInt(ctx->i32, 0, 0), /* idxen */ |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 710 | LLVMConstInt(ctx->i32, glc, 0), |
| 711 | LLVMConstInt(ctx->i32, slc, 0), |
Marek Olšák | 27439df | 2017-02-24 01:20:35 +0100 | [diff] [blame] | 712 | LLVMConstInt(ctx->i32, 0, 0), /* tfe*/ |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 713 | }; |
| 714 | |
| 715 | /* The instruction offset field has 12 bits */ |
Marek Olšák | 8cfdbba | 2017-02-24 20:23:23 +0100 | [diff] [blame] | 716 | assert(voffset || inst_offset < (1 << 12)); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 717 | |
| 718 | /* The intrinsic is overloaded, we need to add a type suffix for overloading to work. */ |
| 719 | unsigned func = CLAMP(num_channels, 1, 3) - 1; |
| 720 | const char *types[] = {"i32", "v2i32", "v4i32"}; |
| 721 | char name[256]; |
| 722 | snprintf(name, sizeof(name), "llvm.SI.tbuffer.store.%s", types[func]); |
| 723 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 724 | ac_build_intrinsic(ctx, name, ctx->voidt, |
| 725 | args, ARRAY_SIZE(args), |
| 726 | AC_FUNC_ATTR_LEGACY); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 727 | } |
| 728 | |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 729 | LLVMValueRef |
| 730 | ac_build_buffer_load(struct ac_llvm_context *ctx, |
| 731 | LLVMValueRef rsrc, |
| 732 | int num_channels, |
| 733 | LLVMValueRef vindex, |
| 734 | LLVMValueRef voffset, |
| 735 | LLVMValueRef soffset, |
| 736 | unsigned inst_offset, |
| 737 | unsigned glc, |
Marek Olšák | e729dc7 | 2017-02-24 17:16:28 +0100 | [diff] [blame] | 738 | unsigned slc, |
Marek Olšák | e019ea8 | 2017-05-19 15:02:34 +0200 | [diff] [blame] | 739 | bool can_speculate, |
| 740 | bool allow_smem) |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 741 | { |
Marek Olšák | e019ea8 | 2017-05-19 15:02:34 +0200 | [diff] [blame] | 742 | LLVMValueRef offset = LLVMConstInt(ctx->i32, inst_offset, 0); |
| 743 | if (voffset) |
| 744 | offset = LLVMBuildAdd(ctx->builder, offset, voffset, ""); |
| 745 | if (soffset) |
| 746 | offset = LLVMBuildAdd(ctx->builder, offset, soffset, ""); |
| 747 | |
| 748 | /* TODO: VI and later generations can use SMEM with GLC=1.*/ |
| 749 | if (allow_smem && !glc && !slc) { |
| 750 | assert(vindex == NULL); |
| 751 | |
| 752 | LLVMValueRef result[4]; |
| 753 | |
| 754 | for (int i = 0; i < num_channels; i++) { |
| 755 | if (i) { |
| 756 | offset = LLVMBuildAdd(ctx->builder, offset, |
| 757 | LLVMConstInt(ctx->i32, 4, 0), ""); |
| 758 | } |
| 759 | LLVMValueRef args[2] = {rsrc, offset}; |
| 760 | result[i] = ac_build_intrinsic(ctx, "llvm.SI.load.const.v4i32", |
| 761 | ctx->f32, args, 2, |
| 762 | AC_FUNC_ATTR_READNONE | |
| 763 | AC_FUNC_ATTR_LEGACY); |
| 764 | } |
| 765 | if (num_channels == 1) |
| 766 | return result[0]; |
| 767 | |
| 768 | if (num_channels == 3) |
| 769 | result[num_channels++] = LLVMGetUndef(ctx->f32); |
| 770 | return ac_build_gather_values(ctx, result, num_channels); |
| 771 | } |
| 772 | |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 773 | unsigned func = CLAMP(num_channels, 1, 3) - 1; |
| 774 | |
Marek Olšák | 12beef0 | 2017-04-25 02:18:10 +0200 | [diff] [blame] | 775 | LLVMValueRef args[] = { |
| 776 | LLVMBuildBitCast(ctx->builder, rsrc, ctx->v4i32, ""), |
| 777 | vindex ? vindex : LLVMConstInt(ctx->i32, 0, 0), |
Marek Olšák | e019ea8 | 2017-05-19 15:02:34 +0200 | [diff] [blame] | 778 | offset, |
Marek Olšák | 12beef0 | 2017-04-25 02:18:10 +0200 | [diff] [blame] | 779 | LLVMConstInt(ctx->i1, glc, 0), |
| 780 | LLVMConstInt(ctx->i1, slc, 0) |
| 781 | }; |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 782 | |
Marek Olšák | 12beef0 | 2017-04-25 02:18:10 +0200 | [diff] [blame] | 783 | LLVMTypeRef types[] = {ctx->f32, LLVMVectorType(ctx->f32, 2), |
| 784 | ctx->v4f32}; |
| 785 | const char *type_names[] = {"f32", "v2f32", "v4f32"}; |
| 786 | char name[256]; |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 787 | |
Marek Olšák | 12beef0 | 2017-04-25 02:18:10 +0200 | [diff] [blame] | 788 | snprintf(name, sizeof(name), "llvm.amdgcn.buffer.load.%s", |
| 789 | type_names[func]); |
| 790 | |
| 791 | return ac_build_intrinsic(ctx, name, types[func], args, |
| 792 | ARRAY_SIZE(args), |
| 793 | /* READNONE means writes can't affect it, while |
| 794 | * READONLY means that writes can affect it. */ |
Marek Olšák | e1942c9 | 2017-05-25 16:13:54 +0200 | [diff] [blame] | 795 | can_speculate && HAVE_LLVM >= 0x0400 ? |
Marek Olšák | 12beef0 | 2017-04-25 02:18:10 +0200 | [diff] [blame] | 796 | AC_FUNC_ATTR_READNONE : |
| 797 | AC_FUNC_ATTR_READONLY); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 798 | } |
| 799 | |
Marek Olšák | 94811dc | 2017-02-25 23:40:52 +0100 | [diff] [blame] | 800 | LLVMValueRef ac_build_buffer_load_format(struct ac_llvm_context *ctx, |
| 801 | LLVMValueRef rsrc, |
| 802 | LLVMValueRef vindex, |
| 803 | LLVMValueRef voffset, |
Marek Olšák | e1942c9 | 2017-05-25 16:13:54 +0200 | [diff] [blame] | 804 | bool can_speculate) |
Marek Olšák | 94811dc | 2017-02-25 23:40:52 +0100 | [diff] [blame] | 805 | { |
Marek Olšák | 12beef0 | 2017-04-25 02:18:10 +0200 | [diff] [blame] | 806 | LLVMValueRef args [] = { |
| 807 | LLVMBuildBitCast(ctx->builder, rsrc, ctx->v4i32, ""), |
Marek Olšák | 94811dc | 2017-02-25 23:40:52 +0100 | [diff] [blame] | 808 | vindex, |
Marek Olšák | 12beef0 | 2017-04-25 02:18:10 +0200 | [diff] [blame] | 809 | voffset, |
| 810 | LLVMConstInt(ctx->i1, 0, 0), /* glc */ |
| 811 | LLVMConstInt(ctx->i1, 0, 0), /* slc */ |
Marek Olšák | 94811dc | 2017-02-25 23:40:52 +0100 | [diff] [blame] | 812 | }; |
Marek Olšák | 12beef0 | 2017-04-25 02:18:10 +0200 | [diff] [blame] | 813 | |
| 814 | return ac_build_intrinsic(ctx, |
| 815 | "llvm.amdgcn.buffer.load.format.v4f32", |
| 816 | ctx->v4f32, args, ARRAY_SIZE(args), |
| 817 | /* READNONE means writes can't affect it, while |
| 818 | * READONLY means that writes can affect it. */ |
Marek Olšák | e1942c9 | 2017-05-25 16:13:54 +0200 | [diff] [blame] | 819 | can_speculate && HAVE_LLVM >= 0x0400 ? |
Marek Olšák | 12beef0 | 2017-04-25 02:18:10 +0200 | [diff] [blame] | 820 | AC_FUNC_ATTR_READNONE : |
| 821 | AC_FUNC_ATTR_READONLY); |
Marek Olšák | 94811dc | 2017-02-25 23:40:52 +0100 | [diff] [blame] | 822 | } |
| 823 | |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 824 | /** |
| 825 | * Set range metadata on an instruction. This can only be used on load and |
| 826 | * call instructions. If you know an instruction can only produce the values |
| 827 | * 0, 1, 2, you would do set_range_metadata(value, 0, 3); |
| 828 | * \p lo is the minimum value inclusive. |
| 829 | * \p hi is the maximum value exclusive. |
| 830 | */ |
| 831 | static void set_range_metadata(struct ac_llvm_context *ctx, |
| 832 | LLVMValueRef value, unsigned lo, unsigned hi) |
| 833 | { |
| 834 | LLVMValueRef range_md, md_args[2]; |
| 835 | LLVMTypeRef type = LLVMTypeOf(value); |
| 836 | LLVMContextRef context = LLVMGetTypeContext(type); |
| 837 | |
| 838 | md_args[0] = LLVMConstInt(type, lo, false); |
| 839 | md_args[1] = LLVMConstInt(type, hi, false); |
| 840 | range_md = LLVMMDNodeInContext(context, md_args, 2); |
| 841 | LLVMSetMetadata(value, ctx->range_md_kind, range_md); |
| 842 | } |
| 843 | |
| 844 | LLVMValueRef |
| 845 | ac_get_thread_id(struct ac_llvm_context *ctx) |
| 846 | { |
| 847 | LLVMValueRef tid; |
| 848 | |
Marek Olšák | 7e1faa7 | 2017-03-05 00:15:31 +0100 | [diff] [blame] | 849 | LLVMValueRef tid_args[2]; |
| 850 | tid_args[0] = LLVMConstInt(ctx->i32, 0xffffffff, false); |
| 851 | tid_args[1] = LLVMConstInt(ctx->i32, 0, false); |
| 852 | tid_args[1] = ac_build_intrinsic(ctx, |
| 853 | "llvm.amdgcn.mbcnt.lo", ctx->i32, |
| 854 | tid_args, 2, AC_FUNC_ATTR_READNONE); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 855 | |
Marek Olšák | 7e1faa7 | 2017-03-05 00:15:31 +0100 | [diff] [blame] | 856 | tid = ac_build_intrinsic(ctx, "llvm.amdgcn.mbcnt.hi", |
| 857 | ctx->i32, tid_args, |
| 858 | 2, AC_FUNC_ATTR_READNONE); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 859 | set_range_metadata(ctx, tid, 0, 64); |
| 860 | return tid; |
| 861 | } |
| 862 | |
| 863 | /* |
| 864 | * SI implements derivatives using the local data store (LDS) |
| 865 | * All writes to the LDS happen in all executing threads at |
| 866 | * the same time. TID is the Thread ID for the current |
| 867 | * thread and is a value between 0 and 63, representing |
| 868 | * the thread's position in the wavefront. |
| 869 | * |
| 870 | * For the pixel shader threads are grouped into quads of four pixels. |
| 871 | * The TIDs of the pixels of a quad are: |
| 872 | * |
| 873 | * +------+------+ |
| 874 | * |4n + 0|4n + 1| |
| 875 | * +------+------+ |
| 876 | * |4n + 2|4n + 3| |
| 877 | * +------+------+ |
| 878 | * |
| 879 | * So, masking the TID with 0xfffffffc yields the TID of the top left pixel |
| 880 | * of the quad, masking with 0xfffffffd yields the TID of the top pixel of |
| 881 | * the current pixel's column, and masking with 0xfffffffe yields the TID |
| 882 | * of the left pixel of the current pixel's row. |
| 883 | * |
| 884 | * Adding 1 yields the TID of the pixel to the right of the left pixel, and |
| 885 | * adding 2 yields the TID of the pixel below the top pixel. |
| 886 | */ |
| 887 | LLVMValueRef |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 888 | ac_build_ddxy(struct ac_llvm_context *ctx, |
| 889 | bool has_ds_bpermute, |
| 890 | uint32_t mask, |
| 891 | int idx, |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 892 | LLVMValueRef val) |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 893 | { |
Dave Airlie | cb6f16d | 2017-08-01 05:10:49 +0100 | [diff] [blame] | 894 | LLVMValueRef tl, trbl, args[2]; |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 895 | LLVMValueRef result; |
| 896 | |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 897 | if (has_ds_bpermute) { |
Dave Airlie | cb6f16d | 2017-08-01 05:10:49 +0100 | [diff] [blame] | 898 | LLVMValueRef thread_id, tl_tid, trbl_tid; |
| 899 | thread_id = ac_get_thread_id(ctx); |
| 900 | |
| 901 | tl_tid = LLVMBuildAnd(ctx->builder, thread_id, |
| 902 | LLVMConstInt(ctx->i32, mask, false), ""); |
| 903 | |
| 904 | trbl_tid = LLVMBuildAdd(ctx->builder, tl_tid, |
| 905 | LLVMConstInt(ctx->i32, idx, false), ""); |
| 906 | |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 907 | args[0] = LLVMBuildMul(ctx->builder, tl_tid, |
| 908 | LLVMConstInt(ctx->i32, 4, false), ""); |
| 909 | args[1] = val; |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 910 | tl = ac_build_intrinsic(ctx, |
| 911 | "llvm.amdgcn.ds.bpermute", ctx->i32, |
Marek Olšák | d5d74fe | 2017-02-26 19:00:44 +0100 | [diff] [blame] | 912 | args, 2, |
| 913 | AC_FUNC_ATTR_READNONE | |
| 914 | AC_FUNC_ATTR_CONVERGENT); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 915 | |
| 916 | args[0] = LLVMBuildMul(ctx->builder, trbl_tid, |
| 917 | LLVMConstInt(ctx->i32, 4, false), ""); |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 918 | trbl = ac_build_intrinsic(ctx, |
| 919 | "llvm.amdgcn.ds.bpermute", ctx->i32, |
Marek Olšák | d5d74fe | 2017-02-26 19:00:44 +0100 | [diff] [blame] | 920 | args, 2, |
| 921 | AC_FUNC_ATTR_READNONE | |
| 922 | AC_FUNC_ATTR_CONVERGENT); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 923 | } else { |
Dave Airlie | cb6f16d | 2017-08-01 05:10:49 +0100 | [diff] [blame] | 924 | uint32_t masks[2]; |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 925 | |
Dave Airlie | cb6f16d | 2017-08-01 05:10:49 +0100 | [diff] [blame] | 926 | switch (mask) { |
| 927 | case AC_TID_MASK_TOP_LEFT: |
| 928 | masks[0] = 0x8000; |
| 929 | if (idx == 1) |
| 930 | masks[1] = 0x8055; |
| 931 | else |
| 932 | masks[1] = 0x80aa; |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 933 | |
Dave Airlie | cb6f16d | 2017-08-01 05:10:49 +0100 | [diff] [blame] | 934 | break; |
| 935 | case AC_TID_MASK_TOP: |
| 936 | masks[0] = 0x8044; |
| 937 | masks[1] = 0x80ee; |
| 938 | break; |
| 939 | case AC_TID_MASK_LEFT: |
| 940 | masks[0] = 0x80a0; |
| 941 | masks[1] = 0x80f5; |
| 942 | break; |
| 943 | } |
| 944 | |
| 945 | args[0] = val; |
| 946 | args[1] = LLVMConstInt(ctx->i32, masks[0], false); |
| 947 | |
| 948 | tl = ac_build_intrinsic(ctx, |
| 949 | "llvm.amdgcn.ds.swizzle", ctx->i32, |
| 950 | args, 2, |
| 951 | AC_FUNC_ATTR_READNONE | |
| 952 | AC_FUNC_ATTR_CONVERGENT); |
| 953 | |
| 954 | args[1] = LLVMConstInt(ctx->i32, masks[1], false); |
| 955 | trbl = ac_build_intrinsic(ctx, |
| 956 | "llvm.amdgcn.ds.swizzle", ctx->i32, |
| 957 | args, 2, |
| 958 | AC_FUNC_ATTR_READNONE | |
| 959 | AC_FUNC_ATTR_CONVERGENT); |
Dave Airlie | 13a28ff | 2017-02-03 10:05:00 +1000 | [diff] [blame] | 960 | } |
| 961 | |
| 962 | tl = LLVMBuildBitCast(ctx->builder, tl, ctx->f32, ""); |
| 963 | trbl = LLVMBuildBitCast(ctx->builder, trbl, ctx->f32, ""); |
| 964 | result = LLVMBuildFSub(ctx->builder, trbl, tl, ""); |
| 965 | return result; |
| 966 | } |
Dave Airlie | f32955b | 2017-02-13 22:08:30 +0000 | [diff] [blame] | 967 | |
| 968 | void |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 969 | ac_build_sendmsg(struct ac_llvm_context *ctx, |
| 970 | uint32_t msg, |
| 971 | LLVMValueRef wave_id) |
Dave Airlie | f32955b | 2017-02-13 22:08:30 +0000 | [diff] [blame] | 972 | { |
| 973 | LLVMValueRef args[2]; |
| 974 | const char *intr_name = (HAVE_LLVM < 0x0400) ? "llvm.SI.sendmsg" : "llvm.amdgcn.s.sendmsg"; |
| 975 | args[0] = LLVMConstInt(ctx->i32, msg, false); |
| 976 | args[1] = wave_id; |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 977 | ac_build_intrinsic(ctx, intr_name, ctx->voidt, args, 2, 0); |
Dave Airlie | f32955b | 2017-02-13 22:08:30 +0000 | [diff] [blame] | 978 | } |
Dave Airlie | cae1ff1 | 2017-02-16 03:42:56 +0000 | [diff] [blame] | 979 | |
| 980 | LLVMValueRef |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 981 | ac_build_imsb(struct ac_llvm_context *ctx, |
| 982 | LLVMValueRef arg, |
| 983 | LLVMTypeRef dst_type) |
Dave Airlie | cae1ff1 | 2017-02-16 03:42:56 +0000 | [diff] [blame] | 984 | { |
Marek Olšák | edd23e0 | 2017-02-17 14:26:07 +0100 | [diff] [blame] | 985 | const char *intr_name = (HAVE_LLVM < 0x0400) ? "llvm.AMDGPU.flbit.i32" : |
| 986 | "llvm.amdgcn.sffbh.i32"; |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 987 | LLVMValueRef msb = ac_build_intrinsic(ctx, intr_name, |
| 988 | dst_type, &arg, 1, |
| 989 | AC_FUNC_ATTR_READNONE); |
Dave Airlie | cae1ff1 | 2017-02-16 03:42:56 +0000 | [diff] [blame] | 990 | |
| 991 | /* The HW returns the last bit index from MSB, but NIR/TGSI wants |
| 992 | * the index from LSB. Invert it by doing "31 - msb". */ |
| 993 | msb = LLVMBuildSub(ctx->builder, LLVMConstInt(ctx->i32, 31, false), |
| 994 | msb, ""); |
| 995 | |
| 996 | LLVMValueRef all_ones = LLVMConstInt(ctx->i32, -1, true); |
| 997 | LLVMValueRef cond = LLVMBuildOr(ctx->builder, |
| 998 | LLVMBuildICmp(ctx->builder, LLVMIntEQ, |
| 999 | arg, LLVMConstInt(ctx->i32, 0, 0), ""), |
| 1000 | LLVMBuildICmp(ctx->builder, LLVMIntEQ, |
| 1001 | arg, all_ones, ""), ""); |
| 1002 | |
| 1003 | return LLVMBuildSelect(ctx->builder, cond, all_ones, msb, ""); |
| 1004 | } |
Dave Airlie | 0ec66b9 | 2017-02-16 03:53:27 +0000 | [diff] [blame] | 1005 | |
| 1006 | LLVMValueRef |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1007 | ac_build_umsb(struct ac_llvm_context *ctx, |
| 1008 | LLVMValueRef arg, |
| 1009 | LLVMTypeRef dst_type) |
Dave Airlie | 0ec66b9 | 2017-02-16 03:53:27 +0000 | [diff] [blame] | 1010 | { |
| 1011 | LLVMValueRef args[2] = { |
| 1012 | arg, |
Marek Olšák | edd23e0 | 2017-02-17 14:26:07 +0100 | [diff] [blame] | 1013 | LLVMConstInt(ctx->i1, 1, 0), |
Dave Airlie | 0ec66b9 | 2017-02-16 03:53:27 +0000 | [diff] [blame] | 1014 | }; |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1015 | LLVMValueRef msb = ac_build_intrinsic(ctx, "llvm.ctlz.i32", |
| 1016 | dst_type, args, ARRAY_SIZE(args), |
| 1017 | AC_FUNC_ATTR_READNONE); |
Dave Airlie | 0ec66b9 | 2017-02-16 03:53:27 +0000 | [diff] [blame] | 1018 | |
| 1019 | /* The HW returns the last bit index from MSB, but TGSI/NIR wants |
| 1020 | * the index from LSB. Invert it by doing "31 - msb". */ |
| 1021 | msb = LLVMBuildSub(ctx->builder, LLVMConstInt(ctx->i32, 31, false), |
| 1022 | msb, ""); |
| 1023 | |
| 1024 | /* check for zero */ |
| 1025 | return LLVMBuildSelect(ctx->builder, |
| 1026 | LLVMBuildICmp(ctx->builder, LLVMIntEQ, arg, |
| 1027 | LLVMConstInt(ctx->i32, 0, 0), ""), |
| 1028 | LLVMConstInt(ctx->i32, -1, true), msb, ""); |
| 1029 | } |
Marek Olšák | 660b55e | 2017-02-16 22:41:16 +0100 | [diff] [blame] | 1030 | |
Nicolai Hähnle | a69afb6 | 2017-06-25 17:56:37 +0200 | [diff] [blame] | 1031 | LLVMValueRef ac_build_umin(struct ac_llvm_context *ctx, LLVMValueRef a, |
| 1032 | LLVMValueRef b) |
| 1033 | { |
| 1034 | LLVMValueRef cmp = LLVMBuildICmp(ctx->builder, LLVMIntULE, a, b, ""); |
| 1035 | return LLVMBuildSelect(ctx->builder, cmp, a, b, ""); |
| 1036 | } |
| 1037 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1038 | LLVMValueRef ac_build_clamp(struct ac_llvm_context *ctx, LLVMValueRef value) |
Marek Olšák | 660b55e | 2017-02-16 22:41:16 +0100 | [diff] [blame] | 1039 | { |
Marek Olšák | 675ef9c | 2017-02-16 22:52:53 +0100 | [diff] [blame] | 1040 | if (HAVE_LLVM >= 0x0500) { |
| 1041 | LLVMValueRef max[2] = { |
| 1042 | value, |
| 1043 | LLVMConstReal(ctx->f32, 0), |
| 1044 | }; |
| 1045 | LLVMValueRef min[2] = { |
| 1046 | LLVMConstReal(ctx->f32, 1), |
| 1047 | }; |
| 1048 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1049 | min[1] = ac_build_intrinsic(ctx, "llvm.maxnum.f32", |
| 1050 | ctx->f32, max, 2, |
| 1051 | AC_FUNC_ATTR_READNONE); |
| 1052 | return ac_build_intrinsic(ctx, "llvm.minnum.f32", |
| 1053 | ctx->f32, min, 2, |
| 1054 | AC_FUNC_ATTR_READNONE); |
Marek Olšák | 675ef9c | 2017-02-16 22:52:53 +0100 | [diff] [blame] | 1055 | } |
| 1056 | |
Marek Olšák | 660b55e | 2017-02-16 22:41:16 +0100 | [diff] [blame] | 1057 | LLVMValueRef args[3] = { |
| 1058 | value, |
| 1059 | LLVMConstReal(ctx->f32, 0), |
| 1060 | LLVMConstReal(ctx->f32, 1), |
| 1061 | }; |
| 1062 | |
Marek Olšák | 7e1faa7 | 2017-03-05 00:15:31 +0100 | [diff] [blame] | 1063 | return ac_build_intrinsic(ctx, "llvm.AMDGPU.clamp.", ctx->f32, args, 3, |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1064 | AC_FUNC_ATTR_READNONE | |
| 1065 | AC_FUNC_ATTR_LEGACY); |
Marek Olšák | 660b55e | 2017-02-16 22:41:16 +0100 | [diff] [blame] | 1066 | } |
Marek Olšák | 369f4a8 | 2017-02-23 02:06:40 +0100 | [diff] [blame] | 1067 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1068 | void ac_build_export(struct ac_llvm_context *ctx, struct ac_export_args *a) |
Marek Olšák | 369f4a8 | 2017-02-23 02:06:40 +0100 | [diff] [blame] | 1069 | { |
| 1070 | LLVMValueRef args[9]; |
| 1071 | |
Marek Olšák | 2b3ebe3 | 2017-02-23 02:15:54 +0100 | [diff] [blame] | 1072 | if (HAVE_LLVM >= 0x0500) { |
| 1073 | args[0] = LLVMConstInt(ctx->i32, a->target, 0); |
| 1074 | args[1] = LLVMConstInt(ctx->i32, a->enabled_channels, 0); |
| 1075 | |
| 1076 | if (a->compr) { |
| 1077 | LLVMTypeRef i16 = LLVMInt16TypeInContext(ctx->context); |
| 1078 | LLVMTypeRef v2i16 = LLVMVectorType(i16, 2); |
| 1079 | |
| 1080 | args[2] = LLVMBuildBitCast(ctx->builder, a->out[0], |
| 1081 | v2i16, ""); |
| 1082 | args[3] = LLVMBuildBitCast(ctx->builder, a->out[1], |
| 1083 | v2i16, ""); |
| 1084 | args[4] = LLVMConstInt(ctx->i1, a->done, 0); |
| 1085 | args[5] = LLVMConstInt(ctx->i1, a->valid_mask, 0); |
| 1086 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1087 | ac_build_intrinsic(ctx, "llvm.amdgcn.exp.compr.v2i16", |
| 1088 | ctx->voidt, args, 6, 0); |
Marek Olšák | 2b3ebe3 | 2017-02-23 02:15:54 +0100 | [diff] [blame] | 1089 | } else { |
| 1090 | args[2] = a->out[0]; |
| 1091 | args[3] = a->out[1]; |
| 1092 | args[4] = a->out[2]; |
| 1093 | args[5] = a->out[3]; |
| 1094 | args[6] = LLVMConstInt(ctx->i1, a->done, 0); |
| 1095 | args[7] = LLVMConstInt(ctx->i1, a->valid_mask, 0); |
| 1096 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1097 | ac_build_intrinsic(ctx, "llvm.amdgcn.exp.f32", |
| 1098 | ctx->voidt, args, 8, 0); |
Marek Olšák | 2b3ebe3 | 2017-02-23 02:15:54 +0100 | [diff] [blame] | 1099 | } |
| 1100 | return; |
| 1101 | } |
| 1102 | |
Marek Olšák | 369f4a8 | 2017-02-23 02:06:40 +0100 | [diff] [blame] | 1103 | args[0] = LLVMConstInt(ctx->i32, a->enabled_channels, 0); |
| 1104 | args[1] = LLVMConstInt(ctx->i32, a->valid_mask, 0); |
| 1105 | args[2] = LLVMConstInt(ctx->i32, a->done, 0); |
| 1106 | args[3] = LLVMConstInt(ctx->i32, a->target, 0); |
| 1107 | args[4] = LLVMConstInt(ctx->i32, a->compr, 0); |
| 1108 | memcpy(args + 5, a->out, sizeof(a->out[0]) * 4); |
| 1109 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1110 | ac_build_intrinsic(ctx, "llvm.SI.export", ctx->voidt, args, 9, |
| 1111 | AC_FUNC_ATTR_LEGACY); |
Marek Olšák | 369f4a8 | 2017-02-23 02:06:40 +0100 | [diff] [blame] | 1112 | } |
Marek Olšák | ad18d7f | 2017-02-23 23:00:19 +0100 | [diff] [blame] | 1113 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1114 | LLVMValueRef ac_build_image_opcode(struct ac_llvm_context *ctx, |
| 1115 | struct ac_image_args *a) |
Marek Olšák | ad18d7f | 2017-02-23 23:00:19 +0100 | [diff] [blame] | 1116 | { |
| 1117 | LLVMTypeRef dst_type; |
| 1118 | LLVMValueRef args[11]; |
| 1119 | unsigned num_args = 0; |
| 1120 | const char *name; |
| 1121 | char intr_name[128], type[64]; |
| 1122 | |
Marek Olšák | 4b2e5b9 | 2017-02-23 23:37:59 +0100 | [diff] [blame] | 1123 | if (HAVE_LLVM >= 0x0400) { |
| 1124 | bool sample = a->opcode == ac_image_sample || |
| 1125 | a->opcode == ac_image_gather4 || |
| 1126 | a->opcode == ac_image_get_lod; |
| 1127 | |
| 1128 | if (sample) |
| 1129 | args[num_args++] = bitcast_to_float(ctx, a->addr); |
| 1130 | else |
| 1131 | args[num_args++] = a->addr; |
| 1132 | |
| 1133 | args[num_args++] = a->resource; |
| 1134 | if (sample) |
| 1135 | args[num_args++] = a->sampler; |
| 1136 | args[num_args++] = LLVMConstInt(ctx->i32, a->dmask, 0); |
| 1137 | if (sample) |
| 1138 | args[num_args++] = LLVMConstInt(ctx->i1, a->unorm, 0); |
| 1139 | args[num_args++] = LLVMConstInt(ctx->i1, 0, 0); /* glc */ |
| 1140 | args[num_args++] = LLVMConstInt(ctx->i1, 0, 0); /* slc */ |
| 1141 | args[num_args++] = LLVMConstInt(ctx->i1, 0, 0); /* lwe */ |
| 1142 | args[num_args++] = LLVMConstInt(ctx->i1, a->da, 0); |
| 1143 | |
| 1144 | switch (a->opcode) { |
| 1145 | case ac_image_sample: |
| 1146 | name = "llvm.amdgcn.image.sample"; |
| 1147 | break; |
| 1148 | case ac_image_gather4: |
| 1149 | name = "llvm.amdgcn.image.gather4"; |
| 1150 | break; |
| 1151 | case ac_image_load: |
| 1152 | name = "llvm.amdgcn.image.load"; |
| 1153 | break; |
| 1154 | case ac_image_load_mip: |
| 1155 | name = "llvm.amdgcn.image.load.mip"; |
| 1156 | break; |
| 1157 | case ac_image_get_lod: |
| 1158 | name = "llvm.amdgcn.image.getlod"; |
| 1159 | break; |
| 1160 | case ac_image_get_resinfo: |
| 1161 | name = "llvm.amdgcn.image.getresinfo"; |
| 1162 | break; |
Samuel Pitoiset | a1c37ff | 2017-04-07 18:44:16 +0200 | [diff] [blame] | 1163 | default: |
| 1164 | unreachable("invalid image opcode"); |
Marek Olšák | 4b2e5b9 | 2017-02-23 23:37:59 +0100 | [diff] [blame] | 1165 | } |
| 1166 | |
| 1167 | ac_build_type_name_for_intr(LLVMTypeOf(args[0]), type, |
| 1168 | sizeof(type)); |
| 1169 | |
| 1170 | snprintf(intr_name, sizeof(intr_name), "%s%s%s%s.v4f32.%s.v8i32", |
| 1171 | name, |
| 1172 | a->compare ? ".c" : "", |
| 1173 | a->bias ? ".b" : |
| 1174 | a->lod ? ".l" : |
| 1175 | a->deriv ? ".d" : |
| 1176 | a->level_zero ? ".lz" : "", |
| 1177 | a->offset ? ".o" : "", |
| 1178 | type); |
| 1179 | |
| 1180 | LLVMValueRef result = |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1181 | ac_build_intrinsic(ctx, intr_name, |
| 1182 | ctx->v4f32, args, num_args, |
| 1183 | AC_FUNC_ATTR_READNONE); |
Marek Olšák | 4b2e5b9 | 2017-02-23 23:37:59 +0100 | [diff] [blame] | 1184 | if (!sample) { |
| 1185 | result = LLVMBuildBitCast(ctx->builder, result, |
| 1186 | ctx->v4i32, ""); |
| 1187 | } |
| 1188 | return result; |
| 1189 | } |
| 1190 | |
Marek Olšák | ad18d7f | 2017-02-23 23:00:19 +0100 | [diff] [blame] | 1191 | args[num_args++] = a->addr; |
| 1192 | args[num_args++] = a->resource; |
| 1193 | |
| 1194 | if (a->opcode == ac_image_load || |
| 1195 | a->opcode == ac_image_load_mip || |
| 1196 | a->opcode == ac_image_get_resinfo) { |
| 1197 | dst_type = ctx->v4i32; |
| 1198 | } else { |
| 1199 | dst_type = ctx->v4f32; |
| 1200 | args[num_args++] = a->sampler; |
| 1201 | } |
| 1202 | |
| 1203 | args[num_args++] = LLVMConstInt(ctx->i32, a->dmask, 0); |
| 1204 | args[num_args++] = LLVMConstInt(ctx->i32, a->unorm, 0); |
| 1205 | args[num_args++] = LLVMConstInt(ctx->i32, 0, 0); /* r128 */ |
| 1206 | args[num_args++] = LLVMConstInt(ctx->i32, a->da, 0); |
| 1207 | args[num_args++] = LLVMConstInt(ctx->i32, 0, 0); /* glc */ |
| 1208 | args[num_args++] = LLVMConstInt(ctx->i32, 0, 0); /* slc */ |
| 1209 | args[num_args++] = LLVMConstInt(ctx->i32, 0, 0); /* tfe */ |
| 1210 | args[num_args++] = LLVMConstInt(ctx->i32, 0, 0); /* lwe */ |
| 1211 | |
| 1212 | switch (a->opcode) { |
| 1213 | case ac_image_sample: |
| 1214 | name = "llvm.SI.image.sample"; |
| 1215 | break; |
| 1216 | case ac_image_gather4: |
| 1217 | name = "llvm.SI.gather4"; |
| 1218 | break; |
| 1219 | case ac_image_load: |
| 1220 | name = "llvm.SI.image.load"; |
| 1221 | break; |
| 1222 | case ac_image_load_mip: |
| 1223 | name = "llvm.SI.image.load.mip"; |
| 1224 | break; |
| 1225 | case ac_image_get_lod: |
| 1226 | name = "llvm.SI.getlod"; |
| 1227 | break; |
| 1228 | case ac_image_get_resinfo: |
| 1229 | name = "llvm.SI.getresinfo"; |
| 1230 | break; |
| 1231 | } |
| 1232 | |
| 1233 | ac_build_type_name_for_intr(LLVMTypeOf(a->addr), type, sizeof(type)); |
| 1234 | snprintf(intr_name, sizeof(intr_name), "%s%s%s%s.%s", |
| 1235 | name, |
| 1236 | a->compare ? ".c" : "", |
| 1237 | a->bias ? ".b" : |
| 1238 | a->lod ? ".l" : |
| 1239 | a->deriv ? ".d" : |
| 1240 | a->level_zero ? ".lz" : "", |
| 1241 | a->offset ? ".o" : "", |
| 1242 | type); |
| 1243 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1244 | return ac_build_intrinsic(ctx, intr_name, |
| 1245 | dst_type, args, num_args, |
| 1246 | AC_FUNC_ATTR_READNONE | |
| 1247 | AC_FUNC_ATTR_LEGACY); |
Marek Olšák | ad18d7f | 2017-02-23 23:00:19 +0100 | [diff] [blame] | 1248 | } |
Marek Olšák | 653ac0b | 2017-02-24 16:38:25 +0100 | [diff] [blame] | 1249 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1250 | LLVMValueRef ac_build_cvt_pkrtz_f16(struct ac_llvm_context *ctx, |
| 1251 | LLVMValueRef args[2]) |
Marek Olšák | 653ac0b | 2017-02-24 16:38:25 +0100 | [diff] [blame] | 1252 | { |
| 1253 | if (HAVE_LLVM >= 0x0500) { |
| 1254 | LLVMTypeRef v2f16 = |
| 1255 | LLVMVectorType(LLVMHalfTypeInContext(ctx->context), 2); |
| 1256 | LLVMValueRef res = |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1257 | ac_build_intrinsic(ctx, "llvm.amdgcn.cvt.pkrtz", |
| 1258 | v2f16, args, 2, |
| 1259 | AC_FUNC_ATTR_READNONE); |
Marek Olšák | 653ac0b | 2017-02-24 16:38:25 +0100 | [diff] [blame] | 1260 | return LLVMBuildBitCast(ctx->builder, res, ctx->i32, ""); |
| 1261 | } |
| 1262 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1263 | return ac_build_intrinsic(ctx, "llvm.SI.packf16", ctx->i32, args, 2, |
| 1264 | AC_FUNC_ATTR_READNONE | |
| 1265 | AC_FUNC_ATTR_LEGACY); |
Marek Olšák | 653ac0b | 2017-02-24 16:38:25 +0100 | [diff] [blame] | 1266 | } |
Marek Olšák | 9c09592 | 2017-02-24 22:44:18 +0100 | [diff] [blame] | 1267 | |
| 1268 | /** |
| 1269 | * KILL, AKA discard in GLSL. |
| 1270 | * |
| 1271 | * \param value kill if value < 0.0 or value == NULL. |
| 1272 | */ |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1273 | void ac_build_kill(struct ac_llvm_context *ctx, LLVMValueRef value) |
Marek Olšák | 9c09592 | 2017-02-24 22:44:18 +0100 | [diff] [blame] | 1274 | { |
| 1275 | if (value) { |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1276 | ac_build_intrinsic(ctx, "llvm.AMDGPU.kill", ctx->voidt, |
| 1277 | &value, 1, AC_FUNC_ATTR_LEGACY); |
Marek Olšák | 9c09592 | 2017-02-24 22:44:18 +0100 | [diff] [blame] | 1278 | } else { |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1279 | ac_build_intrinsic(ctx, "llvm.AMDGPU.kilp", ctx->voidt, |
| 1280 | NULL, 0, AC_FUNC_ATTR_LEGACY); |
Marek Olšák | 9c09592 | 2017-02-24 22:44:18 +0100 | [diff] [blame] | 1281 | } |
| 1282 | } |
Marek Olšák | d4324dd | 2017-02-24 23:06:31 +0100 | [diff] [blame] | 1283 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1284 | LLVMValueRef ac_build_bfe(struct ac_llvm_context *ctx, LLVMValueRef input, |
| 1285 | LLVMValueRef offset, LLVMValueRef width, |
| 1286 | bool is_signed) |
Marek Olšák | d4324dd | 2017-02-24 23:06:31 +0100 | [diff] [blame] | 1287 | { |
| 1288 | LLVMValueRef args[] = { |
| 1289 | input, |
| 1290 | offset, |
| 1291 | width, |
| 1292 | }; |
| 1293 | |
| 1294 | if (HAVE_LLVM >= 0x0500) { |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1295 | return ac_build_intrinsic(ctx, |
| 1296 | is_signed ? "llvm.amdgcn.sbfe.i32" : |
| 1297 | "llvm.amdgcn.ubfe.i32", |
| 1298 | ctx->i32, args, 3, |
| 1299 | AC_FUNC_ATTR_READNONE); |
Marek Olšák | d4324dd | 2017-02-24 23:06:31 +0100 | [diff] [blame] | 1300 | } |
| 1301 | |
Marek Olšák | 7f1446a | 2017-02-26 00:41:37 +0100 | [diff] [blame] | 1302 | return ac_build_intrinsic(ctx, |
| 1303 | is_signed ? "llvm.AMDGPU.bfe.i32" : |
| 1304 | "llvm.AMDGPU.bfe.u32", |
| 1305 | ctx->i32, args, 3, |
| 1306 | AC_FUNC_ATTR_READNONE | |
| 1307 | AC_FUNC_ATTR_LEGACY); |
Marek Olšák | d4324dd | 2017-02-24 23:06:31 +0100 | [diff] [blame] | 1308 | } |
Dave Airlie | 10ae83a | 2017-03-06 08:37:22 +1000 | [diff] [blame] | 1309 | |
| 1310 | void ac_get_image_intr_name(const char *base_name, |
| 1311 | LLVMTypeRef data_type, |
| 1312 | LLVMTypeRef coords_type, |
| 1313 | LLVMTypeRef rsrc_type, |
| 1314 | char *out_name, unsigned out_len) |
| 1315 | { |
| 1316 | char coords_type_name[8]; |
| 1317 | |
| 1318 | ac_build_type_name_for_intr(coords_type, coords_type_name, |
| 1319 | sizeof(coords_type_name)); |
| 1320 | |
| 1321 | if (HAVE_LLVM <= 0x0309) { |
| 1322 | snprintf(out_name, out_len, "%s.%s", base_name, coords_type_name); |
| 1323 | } else { |
| 1324 | char data_type_name[8]; |
| 1325 | char rsrc_type_name[8]; |
| 1326 | |
| 1327 | ac_build_type_name_for_intr(data_type, data_type_name, |
| 1328 | sizeof(data_type_name)); |
| 1329 | ac_build_type_name_for_intr(rsrc_type, rsrc_type_name, |
| 1330 | sizeof(rsrc_type_name)); |
| 1331 | snprintf(out_name, out_len, "%s.%s.%s.%s", base_name, |
| 1332 | data_type_name, coords_type_name, rsrc_type_name); |
| 1333 | } |
| 1334 | } |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1335 | |
| 1336 | #define AC_EXP_TARGET (HAVE_LLVM >= 0x0500 ? 0 : 3) |
| 1337 | #define AC_EXP_OUT0 (HAVE_LLVM >= 0x0500 ? 2 : 5) |
| 1338 | |
Marek Olšák | faa3747 | 2017-04-29 23:47:08 +0200 | [diff] [blame] | 1339 | enum ac_ir_type { |
| 1340 | AC_IR_UNDEF, |
| 1341 | AC_IR_CONST, |
| 1342 | AC_IR_VALUE, |
| 1343 | }; |
| 1344 | |
| 1345 | struct ac_vs_exp_chan |
| 1346 | { |
| 1347 | LLVMValueRef value; |
| 1348 | float const_float; |
| 1349 | enum ac_ir_type type; |
| 1350 | }; |
| 1351 | |
| 1352 | struct ac_vs_exp_inst { |
| 1353 | unsigned offset; |
| 1354 | LLVMValueRef inst; |
| 1355 | struct ac_vs_exp_chan chan[4]; |
| 1356 | }; |
| 1357 | |
| 1358 | struct ac_vs_exports { |
| 1359 | unsigned num; |
| 1360 | struct ac_vs_exp_inst exp[VARYING_SLOT_MAX]; |
| 1361 | }; |
| 1362 | |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1363 | /* Return true if the PARAM export has been eliminated. */ |
| 1364 | static bool ac_eliminate_const_output(uint8_t *vs_output_param_offset, |
| 1365 | uint32_t num_outputs, |
Marek Olšák | faa3747 | 2017-04-29 23:47:08 +0200 | [diff] [blame] | 1366 | struct ac_vs_exp_inst *exp) |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1367 | { |
| 1368 | unsigned i, default_val; /* SPI_PS_INPUT_CNTL_i.DEFAULT_VAL */ |
| 1369 | bool is_zero[4] = {}, is_one[4] = {}; |
| 1370 | |
| 1371 | for (i = 0; i < 4; i++) { |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1372 | /* It's a constant expression. Undef outputs are eliminated too. */ |
Marek Olšák | faa3747 | 2017-04-29 23:47:08 +0200 | [diff] [blame] | 1373 | if (exp->chan[i].type == AC_IR_UNDEF) { |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1374 | is_zero[i] = true; |
| 1375 | is_one[i] = true; |
Marek Olšák | faa3747 | 2017-04-29 23:47:08 +0200 | [diff] [blame] | 1376 | } else if (exp->chan[i].type == AC_IR_CONST) { |
| 1377 | if (exp->chan[i].const_float == 0) |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1378 | is_zero[i] = true; |
Marek Olšák | faa3747 | 2017-04-29 23:47:08 +0200 | [diff] [blame] | 1379 | else if (exp->chan[i].const_float == 1) |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1380 | is_one[i] = true; |
| 1381 | else |
| 1382 | return false; /* other constant */ |
| 1383 | } else |
| 1384 | return false; |
| 1385 | } |
| 1386 | |
| 1387 | /* Only certain combinations of 0 and 1 can be eliminated. */ |
| 1388 | if (is_zero[0] && is_zero[1] && is_zero[2]) |
| 1389 | default_val = is_zero[3] ? 0 : 1; |
| 1390 | else if (is_one[0] && is_one[1] && is_one[2]) |
| 1391 | default_val = is_zero[3] ? 2 : 3; |
| 1392 | else |
| 1393 | return false; |
| 1394 | |
| 1395 | /* The PARAM export can be represented as DEFAULT_VAL. Kill it. */ |
Marek Olšák | faa3747 | 2017-04-29 23:47:08 +0200 | [diff] [blame] | 1396 | LLVMInstructionEraseFromParent(exp->inst); |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1397 | |
| 1398 | /* Change OFFSET to DEFAULT_VAL. */ |
| 1399 | for (i = 0; i < num_outputs; i++) { |
Marek Olšák | faa3747 | 2017-04-29 23:47:08 +0200 | [diff] [blame] | 1400 | if (vs_output_param_offset[i] == exp->offset) { |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1401 | vs_output_param_offset[i] = |
| 1402 | AC_EXP_PARAM_DEFAULT_VAL_0000 + default_val; |
| 1403 | break; |
| 1404 | } |
| 1405 | } |
| 1406 | return true; |
| 1407 | } |
| 1408 | |
Marek Olšák | b087154 | 2017-04-29 23:56:03 +0200 | [diff] [blame] | 1409 | static bool ac_eliminate_duplicated_output(uint8_t *vs_output_param_offset, |
| 1410 | uint32_t num_outputs, |
| 1411 | struct ac_vs_exports *processed, |
| 1412 | struct ac_vs_exp_inst *exp) |
| 1413 | { |
| 1414 | unsigned p, copy_back_channels = 0; |
| 1415 | |
| 1416 | /* See if the output is already in the list of processed outputs. |
| 1417 | * The LLVMValueRef comparison relies on SSA. |
| 1418 | */ |
| 1419 | for (p = 0; p < processed->num; p++) { |
| 1420 | bool different = false; |
| 1421 | |
| 1422 | for (unsigned j = 0; j < 4; j++) { |
| 1423 | struct ac_vs_exp_chan *c1 = &processed->exp[p].chan[j]; |
| 1424 | struct ac_vs_exp_chan *c2 = &exp->chan[j]; |
| 1425 | |
| 1426 | /* Treat undef as a match. */ |
| 1427 | if (c2->type == AC_IR_UNDEF) |
| 1428 | continue; |
| 1429 | |
| 1430 | /* If c1 is undef but c2 isn't, we can copy c2 to c1 |
| 1431 | * and consider the instruction duplicated. |
| 1432 | */ |
| 1433 | if (c1->type == AC_IR_UNDEF) { |
| 1434 | copy_back_channels |= 1 << j; |
| 1435 | continue; |
| 1436 | } |
| 1437 | |
| 1438 | /* Test whether the channels are not equal. */ |
| 1439 | if (c1->type != c2->type || |
| 1440 | (c1->type == AC_IR_CONST && |
| 1441 | c1->const_float != c2->const_float) || |
| 1442 | (c1->type == AC_IR_VALUE && |
| 1443 | c1->value != c2->value)) { |
| 1444 | different = true; |
| 1445 | break; |
| 1446 | } |
| 1447 | } |
| 1448 | if (!different) |
| 1449 | break; |
| 1450 | |
| 1451 | copy_back_channels = 0; |
| 1452 | } |
| 1453 | if (p == processed->num) |
| 1454 | return false; |
| 1455 | |
| 1456 | /* If a match was found, but the matching export has undef where the new |
| 1457 | * one has a normal value, copy the normal value to the undef channel. |
| 1458 | */ |
| 1459 | struct ac_vs_exp_inst *match = &processed->exp[p]; |
| 1460 | |
| 1461 | while (copy_back_channels) { |
| 1462 | unsigned chan = u_bit_scan(©_back_channels); |
| 1463 | |
| 1464 | assert(match->chan[chan].type == AC_IR_UNDEF); |
| 1465 | LLVMSetOperand(match->inst, AC_EXP_OUT0 + chan, |
| 1466 | exp->chan[chan].value); |
| 1467 | match->chan[chan] = exp->chan[chan]; |
| 1468 | } |
| 1469 | |
| 1470 | /* The PARAM export is duplicated. Kill it. */ |
| 1471 | LLVMInstructionEraseFromParent(exp->inst); |
| 1472 | |
| 1473 | /* Change OFFSET to the matching export. */ |
| 1474 | for (unsigned i = 0; i < num_outputs; i++) { |
| 1475 | if (vs_output_param_offset[i] == exp->offset) { |
| 1476 | vs_output_param_offset[i] = match->offset; |
| 1477 | break; |
| 1478 | } |
| 1479 | } |
| 1480 | return true; |
| 1481 | } |
| 1482 | |
Marek Olšák | 7647e90 | 2017-04-29 23:53:08 +0200 | [diff] [blame] | 1483 | void ac_optimize_vs_outputs(struct ac_llvm_context *ctx, |
| 1484 | LLVMValueRef main_fn, |
| 1485 | uint8_t *vs_output_param_offset, |
| 1486 | uint32_t num_outputs, |
| 1487 | uint8_t *num_param_exports) |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1488 | { |
| 1489 | LLVMBasicBlockRef bb; |
| 1490 | bool removed_any = false; |
| 1491 | struct ac_vs_exports exports; |
| 1492 | |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1493 | exports.num = 0; |
| 1494 | |
| 1495 | /* Process all LLVM instructions. */ |
| 1496 | bb = LLVMGetFirstBasicBlock(main_fn); |
| 1497 | while (bb) { |
| 1498 | LLVMValueRef inst = LLVMGetFirstInstruction(bb); |
| 1499 | |
| 1500 | while (inst) { |
| 1501 | LLVMValueRef cur = inst; |
| 1502 | inst = LLVMGetNextInstruction(inst); |
Marek Olšák | faa3747 | 2017-04-29 23:47:08 +0200 | [diff] [blame] | 1503 | struct ac_vs_exp_inst exp; |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1504 | |
| 1505 | if (LLVMGetInstructionOpcode(cur) != LLVMCall) |
| 1506 | continue; |
| 1507 | |
| 1508 | LLVMValueRef callee = ac_llvm_get_called_value(cur); |
| 1509 | |
| 1510 | if (!ac_llvm_is_function(callee)) |
| 1511 | continue; |
| 1512 | |
| 1513 | const char *name = LLVMGetValueName(callee); |
| 1514 | unsigned num_args = LLVMCountParams(callee); |
| 1515 | |
| 1516 | /* Check if this is an export instruction. */ |
| 1517 | if ((num_args != 9 && num_args != 8) || |
| 1518 | (strcmp(name, "llvm.SI.export") && |
| 1519 | strcmp(name, "llvm.amdgcn.exp.f32"))) |
| 1520 | continue; |
| 1521 | |
| 1522 | LLVMValueRef arg = LLVMGetOperand(cur, AC_EXP_TARGET); |
| 1523 | unsigned target = LLVMConstIntGetZExtValue(arg); |
| 1524 | |
| 1525 | if (target < V_008DFC_SQ_EXP_PARAM) |
| 1526 | continue; |
| 1527 | |
| 1528 | target -= V_008DFC_SQ_EXP_PARAM; |
| 1529 | |
Marek Olšák | faa3747 | 2017-04-29 23:47:08 +0200 | [diff] [blame] | 1530 | /* Parse the instruction. */ |
| 1531 | memset(&exp, 0, sizeof(exp)); |
| 1532 | exp.offset = target; |
| 1533 | exp.inst = cur; |
| 1534 | |
| 1535 | for (unsigned i = 0; i < 4; i++) { |
| 1536 | LLVMValueRef v = LLVMGetOperand(cur, AC_EXP_OUT0 + i); |
| 1537 | |
| 1538 | exp.chan[i].value = v; |
| 1539 | |
| 1540 | if (LLVMIsUndef(v)) { |
| 1541 | exp.chan[i].type = AC_IR_UNDEF; |
| 1542 | } else if (LLVMIsAConstantFP(v)) { |
| 1543 | LLVMBool loses_info; |
| 1544 | exp.chan[i].type = AC_IR_CONST; |
| 1545 | exp.chan[i].const_float = |
| 1546 | LLVMConstRealGetDouble(v, &loses_info); |
| 1547 | } else { |
| 1548 | exp.chan[i].type = AC_IR_VALUE; |
| 1549 | } |
| 1550 | } |
| 1551 | |
Marek Olšák | b087154 | 2017-04-29 23:56:03 +0200 | [diff] [blame] | 1552 | /* Eliminate constant and duplicated PARAM exports. */ |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1553 | if (ac_eliminate_const_output(vs_output_param_offset, |
Marek Olšák | b087154 | 2017-04-29 23:56:03 +0200 | [diff] [blame] | 1554 | num_outputs, &exp) || |
| 1555 | ac_eliminate_duplicated_output(vs_output_param_offset, |
| 1556 | num_outputs, &exports, |
| 1557 | &exp)) { |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1558 | removed_any = true; |
| 1559 | } else { |
Marek Olšák | faa3747 | 2017-04-29 23:47:08 +0200 | [diff] [blame] | 1560 | exports.exp[exports.num++] = exp; |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1561 | } |
| 1562 | } |
| 1563 | bb = LLVMGetNextBasicBlock(bb); |
| 1564 | } |
| 1565 | |
| 1566 | /* Remove holes in export memory due to removed PARAM exports. |
| 1567 | * This is done by renumbering all PARAM exports. |
| 1568 | */ |
| 1569 | if (removed_any) { |
Marek Olšák | 34bc470 | 2017-05-08 16:37:26 +0200 | [diff] [blame] | 1570 | uint8_t old_offset[VARYING_SLOT_MAX]; |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1571 | unsigned out, i; |
| 1572 | |
| 1573 | /* Make a copy of the offsets. We need the old version while |
| 1574 | * we are modifying some of them. */ |
Marek Olšák | 34bc470 | 2017-05-08 16:37:26 +0200 | [diff] [blame] | 1575 | memcpy(old_offset, vs_output_param_offset, |
| 1576 | sizeof(old_offset)); |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1577 | |
| 1578 | for (i = 0; i < exports.num; i++) { |
Marek Olšák | faa3747 | 2017-04-29 23:47:08 +0200 | [diff] [blame] | 1579 | unsigned offset = exports.exp[i].offset; |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1580 | |
Marek Olšák | 34bc470 | 2017-05-08 16:37:26 +0200 | [diff] [blame] | 1581 | /* Update vs_output_param_offset. Multiple outputs can |
| 1582 | * have the same offset. |
| 1583 | */ |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1584 | for (out = 0; out < num_outputs; out++) { |
Marek Olšák | 34bc470 | 2017-05-08 16:37:26 +0200 | [diff] [blame] | 1585 | if (old_offset[out] == offset) |
| 1586 | vs_output_param_offset[out] = i; |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1587 | } |
Marek Olšák | 34bc470 | 2017-05-08 16:37:26 +0200 | [diff] [blame] | 1588 | |
| 1589 | /* Change the PARAM offset in the instruction. */ |
| 1590 | LLVMSetOperand(exports.exp[i].inst, AC_EXP_TARGET, |
| 1591 | LLVMConstInt(ctx->i32, |
| 1592 | V_008DFC_SQ_EXP_PARAM + i, 0)); |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1593 | } |
Marek Olšák | 34bc470 | 2017-05-08 16:37:26 +0200 | [diff] [blame] | 1594 | *num_param_exports = exports.num; |
Dave Airlie | e265917 | 2017-04-25 23:33:29 +0100 | [diff] [blame] | 1595 | } |
| 1596 | } |