src/gallium/drivers/radeonsi/radeonsi_shader.c

   1
   2 /*
   3  * Copyright 2012 Advanced Micro Devices, Inc.
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the "Software"),
   7  * to deal in the Software without restriction, including without limitation
   8  * on the rights to use, copy, modify, merge, publish, distribute, sub
   9  * license, and/or sell copies of the Software, and to permit persons to whom
  10  * the Software is furnished to do so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice (including the next
  13  * paragraph) shall be included in all copies or substantial portions of the
  14  * Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
  19  * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM,
  20  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  21  * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
  22  * USE OR OTHER DEALINGS IN THE SOFTWARE.
  23  *
  24  * Authors:
  25  *      Tom Stellard <thomas.stellard@amd.com>
  26  *      Michel Dänzer <michel.daenzer@amd.com>
  27  *      Christian König <christian.koenig@amd.com>
  28  */
  29
  30 #include "gallivm/lp_bld_tgsi_action.h"
  31 #include "gallivm/lp_bld_const.h"
  32 #include "gallivm/lp_bld_gather.h"
  33 #include "gallivm/lp_bld_intr.h"
  34 #include "gallivm/lp_bld_tgsi.h"
  35 #include "radeon_llvm.h"
  36 #include "radeon_llvm_emit.h"
  37 #include "tgsi/tgsi_info.h"
  38 #include "tgsi/tgsi_parse.h"
  39 #include "tgsi/tgsi_scan.h"
  40 #include "tgsi/tgsi_dump.h"
  41
  42 #include "radeonsi_pipe.h"
  43 #include "radeonsi_shader.h"
  44 #include "si_state.h"
  45 #include "sid.h"
  46
  47 #include <assert.h>
  48 #include <errno.h>
  49 #include <stdio.h>
  50
  51 /*
  52 static ps_remap_inputs(
  53         struct tgsi_llvm_context * tl_ctx,
  54         unsigned tgsi_index,
  55         unsigned tgsi_chan)
  56 {
  57         :
  58 }
  59
  60 struct si_input
  61 {
  62         struct list_head head;
  63         unsigned tgsi_index;
  64         unsigned tgsi_chan;
  65         unsigned order;
  66 };
  67 */
  68
  69
  70 struct si_shader_context
  71 {
  72         struct radeon_llvm_context radeon_bld;
  73         struct r600_context *rctx;
  74         struct tgsi_parse_context parse;
  75         struct tgsi_token * tokens;
  76         struct si_pipe_shader *shader;
  77         unsigned type; /* TGSI_PROCESSOR_* specifies the type of shader. */
  78 /*      unsigned num_inputs; */
  79 /*      struct list_head inputs; */
  80 /*      unsigned * input_mappings *//* From TGSI to SI hw */
  81 /*      struct tgsi_shader_info info;*/
  82 };
  83
  84 static struct si_shader_context * si_shader_context(
  85         struct lp_build_tgsi_context * bld_base)
  86 {
  87         return (struct si_shader_context *)bld_base;
  88 }
  89
  90
  91 #define PERSPECTIVE_BASE 0
  92 #define LINEAR_BASE 9
  93
  94 #define SAMPLE_OFFSET 0
  95 #define CENTER_OFFSET 2
  96 #define CENTROID_OFSET 4
  97
  98 #define USE_SGPR_MAX_SUFFIX_LEN 5
  99 #define CONST_ADDR_SPACE 2
 100 #define USER_SGPR_ADDR_SPACE 8
 101
 102 enum sgpr_type {
 103         SGPR_CONST_PTR_F32,
 104         SGPR_CONST_PTR_V4I32,
 105         SGPR_CONST_PTR_V8I32,
 106         SGPR_I32,
 107         SGPR_I64
 108 };
 109
 110 /**
 111  * Build an LLVM bytecode indexed load using LLVMBuildGEP + LLVMBuildLoad
 112  *
 113  * @param offset The offset parameter specifies the number of
 114  * elements to offset, not the number of bytes or dwords.  An element is the
 115  * the type pointed to by the base_ptr parameter (e.g. int is the element of
 116  * an int* pointer)
 117  *
 118  * When LLVM lowers the load instruction, it will convert the element offset
 119  * into a dword offset automatically.
 120  *
 121  */
 122 static LLVMValueRef build_indexed_load(
 123         struct gallivm_state * gallivm,
 124         LLVMValueRef base_ptr,
 125         LLVMValueRef offset)
 126 {
 127         LLVMValueRef computed_ptr = LLVMBuildGEP(
 128                 gallivm->builder, base_ptr, &offset, 1, "");
 129
 130         return LLVMBuildLoad(gallivm->builder, computed_ptr, "");
 131 }
 132
 133 /**
 134  * Load a value stored in one of the user SGPRs
 135  *
 136  * @param sgpr This is the sgpr to load the value from.  If you need to load a
 137  * value that is stored in consecutive SGPR registers (e.g. a 64-bit pointer),
 138  * then you should pass the index of the first SGPR that holds the value.  For
 139  * example, if you want to load a pointer that is stored in SGPRs 2 and 3, then
 140  * use pass 2 for the sgpr parameter.
 141  *
 142  * The value of the sgpr parameter must also be aligned to the width of the type
 143  * being loaded, so that the sgpr parameter is divisible by the dword width of the
 144  * type.  For example, if the value being loaded is two dwords wide, then the sgpr
 145  * parameter must be divisible by two.
 146  */
 147 static LLVMValueRef use_sgpr(
 148         struct gallivm_state * gallivm,
 149         enum sgpr_type type,
 150         unsigned sgpr)
 151 {
 152         LLVMValueRef sgpr_index;
 153         LLVMTypeRef ret_type;
 154         LLVMValueRef ptr;
 155
 156         sgpr_index = lp_build_const_int32(gallivm, sgpr);
 157
 158         switch (type) {
 159         case SGPR_CONST_PTR_F32:
 160                 assert(sgpr % 2 == 0);
 161                 ret_type = LLVMFloatTypeInContext(gallivm->context);
 162                 ret_type = LLVMPointerType(ret_type, CONST_ADDR_SPACE);
 163                 break;
 164
 165         case SGPR_I32:
 166                 ret_type = LLVMInt32TypeInContext(gallivm->context);
 167                 break;
 168
 169         case SGPR_I64:
 170                 assert(sgpr % 2 == 0);
 171                 ret_type= LLVMInt64TypeInContext(gallivm->context);
 172                 break;
 173
 174         case SGPR_CONST_PTR_V4I32:
 175                 assert(sgpr % 2 == 0);
 176                 ret_type = LLVMInt32TypeInContext(gallivm->context);
 177                 ret_type = LLVMVectorType(ret_type, 4);
 178                 ret_type = LLVMPointerType(ret_type, CONST_ADDR_SPACE);
 179                 break;
 180
 181         case SGPR_CONST_PTR_V8I32:
 182                 assert(sgpr % 2 == 0);
 183                 ret_type = LLVMInt32TypeInContext(gallivm->context);
 184                 ret_type = LLVMVectorType(ret_type, 8);
 185                 ret_type = LLVMPointerType(ret_type, CONST_ADDR_SPACE);
 186                 break;
 187
 188         default:
 189                 assert(!"Unsupported SGPR type in use_sgpr()");
 190                 return NULL;
 191         }
 192
 193         ret_type = LLVMPointerType(ret_type, USER_SGPR_ADDR_SPACE);
 194         ptr = LLVMBuildIntToPtr(gallivm->builder, sgpr_index, ret_type, "");
 195         return LLVMBuildLoad(gallivm->builder, ptr, "");
 196 }
 197
 198 static void declare_input_vs(
 199         struct si_shader_context * si_shader_ctx,
 200         unsigned input_index,
 201         const struct tgsi_full_declaration *decl)
 202 {
 203         LLVMValueRef t_list_ptr;
 204         LLVMValueRef t_offset;
 205         LLVMValueRef t_list;
 206         LLVMValueRef attribute_offset;
 207         LLVMValueRef buffer_index_reg;
 208         LLVMValueRef args[3];
 209         LLVMTypeRef vec4_type;
 210         LLVMValueRef input;
 211         struct lp_build_context * uint = &si_shader_ctx->radeon_bld.soa.bld_base.uint_bld;
 212         struct lp_build_context * base = &si_shader_ctx->radeon_bld.soa.bld_base.base;
 213         struct r600_context *rctx = si_shader_ctx->rctx;
 214         //struct pipe_vertex_element *velem = &rctx->vertex_elements->elements[input_index];
 215         unsigned chan;
 216
 217         /* Load the T list */
 218         /* XXX: Communicate with the rest of the driver about which SGPR the T#
 219          * list pointer is going to be stored in.  Hard code to SGPR[6:7] for
 220          * now */
 221         t_list_ptr = use_sgpr(base->gallivm, SGPR_CONST_PTR_V4I32, 6);
 222
 223         t_offset = lp_build_const_int32(base->gallivm, input_index);
 224
 225         t_list = build_indexed_load(base->gallivm, t_list_ptr, t_offset);
 226
 227         /* Build the attribute offset */
 228         attribute_offset = lp_build_const_int32(base->gallivm, 0);
 229
 230         /* Load the buffer index is always, which is always stored in VGPR0
 231          * for Vertex Shaders */
 232         buffer_index_reg = build_intrinsic(base->gallivm->builder,
 233                 "llvm.SI.vs.load.buffer.index", uint->elem_type, NULL, 0,
 234                 LLVMReadNoneAttribute);
 235
 236         vec4_type = LLVMVectorType(base->elem_type, 4);
 237         args[0] = t_list;
 238         args[1] = attribute_offset;
 239         args[2] = buffer_index_reg;
 240         input = lp_build_intrinsic(base->gallivm->builder,
 241                 "llvm.SI.vs.load.input", vec4_type, args, 3);
 242
 243         /* Break up the vec4 into individual components */
 244         for (chan = 0; chan < 4; chan++) {
 245                 LLVMValueRef llvm_chan = lp_build_const_int32(base->gallivm, chan);
 246                 /* XXX: Use a helper function for this.  There is one in
 247                  * tgsi_llvm.c. */
 248                 si_shader_ctx->radeon_bld.inputs[radeon_llvm_reg_index_soa(input_index, chan)] =
 249                                 LLVMBuildExtractElement(base->gallivm->builder,
 250                                 input, llvm_chan, "");
 251         }
 252 }
 253
 254 static void declare_input_fs(
 255         struct si_shader_context * si_shader_ctx,
 256         unsigned input_index,
 257         const struct tgsi_full_declaration *decl)
 258 {
 259         const char * intr_name;
 260         unsigned chan;
 261         struct lp_build_context * base =
 262                                 &si_shader_ctx->radeon_bld.soa.bld_base.base;
 263         struct gallivm_state * gallivm = base->gallivm;
 264
 265         /* This value is:
 266          * [15:0] NewPrimMask (Bit mask for each quad.  It is set it the
 267          *                     quad begins a new primitive.  Bit 0 always needs
 268          *                     to be unset)
 269          * [32:16] ParamOffset
 270          *
 271          */
 272         /* XXX: This register number must be identical to the S_00B02C_USER_SGPR
 273          * register field value
 274          */
 275         LLVMValueRef params = use_sgpr(base->gallivm, SGPR_I32, 6);
 276
 277
 278         /* XXX: Is this the input_index? */
 279         LLVMValueRef attr_number = lp_build_const_int32(gallivm, input_index);
 280
 281         /* XXX: Handle all possible interpolation modes */
 282         switch (decl->Interp.Interpolate) {
 283         case TGSI_INTERPOLATE_COLOR:
 284                 /* XXX: Flat shading hangs the GPU */
 285                 if (si_shader_ctx->rctx->queued.named.rasterizer &&
 286                     si_shader_ctx->rctx->queued.named.rasterizer->flatshade) {
 287 #if 0
 288                         intr_name = "llvm.SI.fs.interp.constant";
 289 #else
 290                         intr_name = "llvm.SI.fs.interp.linear.center";
 291 #endif
 292                 } else {
 293                         if (decl->Interp.Centroid)
 294                                 intr_name = "llvm.SI.fs.interp.persp.centroid";
 295                         else
 296                                 intr_name = "llvm.SI.fs.interp.persp.center";
 297                 }
 298                 break;
 299         case TGSI_INTERPOLATE_CONSTANT:
 300                 /* XXX: Flat shading hangs the GPU */
 301 #if 0
 302                 intr_name = "llvm.SI.fs.interp.constant";
 303                 break;
 304 #endif
 305         case TGSI_INTERPOLATE_LINEAR:
 306                 if (decl->Interp.Centroid)
 307                         intr_name = "llvm.SI.fs.interp.linear.centroid";
 308                 else
 309                         intr_name = "llvm.SI.fs.interp.linear.center";
 310                 break;
 311         case TGSI_INTERPOLATE_PERSPECTIVE:
 312                 if (decl->Interp.Centroid)
 313                         intr_name = "llvm.SI.fs.interp.persp.centroid";
 314                 else
 315                         intr_name = "llvm.SI.fs.interp.persp.center";
 316                 break;
 317         default:
 318                 fprintf(stderr, "Warning: Unhandled interpolation mode.\n");
 319                 return;
 320         }
 321
 322         /* XXX: Could there be more than TGSI_NUM_CHANNELS (4) ? */
 323         for (chan = 0; chan < TGSI_NUM_CHANNELS; chan++) {
 324                 LLVMValueRef args[3];
 325                 LLVMValueRef llvm_chan = lp_build_const_int32(gallivm, chan);
 326                 unsigned soa_index = radeon_llvm_reg_index_soa(input_index, chan);
 327                 LLVMTypeRef input_type = LLVMFloatTypeInContext(gallivm->context);
 328                 args[0] = llvm_chan;
 329                 args[1] = attr_number;
 330                 args[2] = params;
 331                 si_shader_ctx->radeon_bld.inputs[soa_index] =
 332                         lp_build_intrinsic(gallivm->builder, intr_name,
 333                                                 input_type, args, 3);
 334         }
 335 }
 336
 337 static void declare_input(
 338         struct radeon_llvm_context * radeon_bld,
 339         unsigned input_index,
 340         const struct tgsi_full_declaration *decl)
 341 {
 342         struct si_shader_context * si_shader_ctx =
 343                                 si_shader_context(&radeon_bld->soa.bld_base);
 344         if (si_shader_ctx->type == TGSI_PROCESSOR_VERTEX) {
 345                 declare_input_vs(si_shader_ctx, input_index, decl);
 346         } else if (si_shader_ctx->type == TGSI_PROCESSOR_FRAGMENT) {
 347                 declare_input_fs(si_shader_ctx, input_index, decl);
 348         } else {
 349                 fprintf(stderr, "Warning: Unsupported shader type,\n");
 350         }
 351 }
 352
 353 static LLVMValueRef fetch_constant(
 354         struct lp_build_tgsi_context * bld_base,
 355         const struct tgsi_full_src_register *reg,
 356         enum tgsi_opcode_type type,
 357         unsigned swizzle)
 358 {
 359         struct lp_build_context * base = &bld_base->base;
 360
 361         LLVMValueRef const_ptr;
 362         LLVMValueRef offset;
 363         LLVMValueRef load;
 364
 365         /* XXX: Assume the pointer to the constant buffer is being stored in
 366          * SGPR[0:1] */
 367         const_ptr = use_sgpr(base->gallivm, SGPR_CONST_PTR_F32, 0);
 368
 369         /* XXX: This assumes that the constant buffer is not packed, so
 370          * CONST[0].x will have an offset of 0 and CONST[1].x will have an
 371          * offset of 4. */
 372         offset = lp_build_const_int32(base->gallivm,
 373                                         (reg->Register.Index * 4) + swizzle);
 374
 375         load = build_indexed_load(base->gallivm, const_ptr, offset);
 376         return bitcast(bld_base, type, load);
 377 }
 378
 379 /* Initialize arguments for the shader export intrinsic */
 380 static void si_llvm_init_export_args(struct lp_build_tgsi_context *bld_base,
 381                                      struct tgsi_full_declaration *d,
 382                                      unsigned index,
 383                                      unsigned target,
 384                                      LLVMValueRef *args)
 385 {
 386         struct si_shader_context *si_shader_ctx = si_shader_context(bld_base);
 387         struct lp_build_context *uint =
 388                                 &si_shader_ctx->radeon_bld.soa.bld_base.uint_bld;
 389         struct lp_build_context *base = &bld_base->base;
 390         unsigned compressed = 0;
 391         unsigned chan;
 392
 393         for (chan = 0; chan < 4; chan++ ) {
 394                 LLVMValueRef out_ptr =
 395                         si_shader_ctx->radeon_bld.soa.outputs[index][chan];
 396                 /* +5 because the first output value will be
 397                  * the 6th argument to the intrinsic. */
 398                 args[chan + 5] = LLVMBuildLoad(base->gallivm->builder,
 399                                                out_ptr, "");
 400         }
 401
 402         /* XXX: This controls which components of the output
 403          * registers actually get exported. (e.g bit 0 means export
 404          * X component, bit 1 means export Y component, etc.)  I'm
 405          * hard coding this to 0xf for now.  In the future, we might
 406          * want to do something else. */
 407         args[0] = lp_build_const_int32(base->gallivm, 0xf);
 408
 409         /* Specify whether the EXEC mask represents the valid mask */
 410         args[1] = uint->zero;
 411
 412         /* Specify whether this is the last export */
 413         args[2] = uint->zero;
 414
 415         /* Specify the target we are exporting */
 416         args[3] = lp_build_const_int32(base->gallivm, target);
 417
 418         /* Set COMPR flag */
 419         args[4] = uint->zero;
 420
 421         /* XXX: We probably need to keep track of the output
 422          * values, so we know what we are passing to the next
 423          * stage. */
 424 }
 425
 426 /* XXX: This is partially implemented for VS only at this point.  It is not complete */
 427 static void si_llvm_emit_epilogue(struct lp_build_tgsi_context * bld_base)
 428 {
 429         struct si_shader_context * si_shader_ctx = si_shader_context(bld_base);
 430         struct si_shader * shader = &si_shader_ctx->shader->shader;
 431         struct lp_build_context * base = &bld_base->base;
 432         struct lp_build_context * uint =
 433                                 &si_shader_ctx->radeon_bld.soa.bld_base.uint_bld;
 434         struct tgsi_parse_context *parse = &si_shader_ctx->parse;
 435         LLVMValueRef last_args[9] = { 0 };
 436         unsigned color_count = 0;
 437         unsigned param_count = 0;
 438
 439         while (!tgsi_parse_end_of_tokens(parse)) {
 440                 struct tgsi_full_declaration *d =
 441                                         &parse->FullToken.FullDeclaration;
 442                 LLVMValueRef args[9];
 443                 unsigned target;
 444                 unsigned index;
 445                 int i;
 446
 447                 tgsi_parse_token(parse);
 448                 if (parse->FullToken.Token.Type != TGSI_TOKEN_TYPE_DECLARATION)
 449                         continue;
 450
 451                 switch (d->Declaration.File) {
 452                 case TGSI_FILE_INPUT:
 453                         i = shader->ninput++;
 454                         shader->input[i].name = d->Semantic.Name;
 455                         shader->input[i].sid = d->Semantic.Index;
 456                         shader->input[i].interpolate = d->Interp.Interpolate;
 457                         shader->input[i].centroid = d->Interp.Centroid;
 458                         continue;
 459
 460                 case TGSI_FILE_OUTPUT:
 461                         i = shader->noutput++;
 462                         shader->output[i].name = d->Semantic.Name;
 463                         shader->output[i].sid = d->Semantic.Index;
 464                         shader->output[i].interpolate = d->Interp.Interpolate;
 465                         break;
 466
 467                 default:
 468                         continue;
 469                 }
 470
 471                 for (index = d->Range.First; index <= d->Range.Last; index++) {
 472                         /* Select the correct target */
 473                         switch(d->Semantic.Name) {
 474                         case TGSI_SEMANTIC_POSITION:
 475                                 target = V_008DFC_SQ_EXP_POS;
 476                                 break;
 477                         case TGSI_SEMANTIC_COLOR:
 478                                 if (si_shader_ctx->type == TGSI_PROCESSOR_VERTEX) {
 479                                         target = V_008DFC_SQ_EXP_PARAM + param_count;
 480                                         shader->output[i].param_offset = param_count;
 481                                         param_count++;
 482                                 } else {
 483                                         target = V_008DFC_SQ_EXP_MRT + color_count;
 484                                         color_count++;
 485                                 }
 486                                 break;
 487                         case TGSI_SEMANTIC_GENERIC:
 488                                 target = V_008DFC_SQ_EXP_PARAM + param_count;
 489                                 shader->output[i].param_offset = param_count;
 490                                 param_count++;
 491                                 break;
 492                         default:
 493                                 target = 0;
 494                                 fprintf(stderr,
 495                                         "Warning: SI unhandled output type:%d\n",
 496                                         d->Semantic.Name);
 497                         }
 498
 499                         si_llvm_init_export_args(bld_base, d, index, target, args);
 500
 501                         if (si_shader_ctx->type == TGSI_PROCESSOR_VERTEX ?
 502                             (d->Semantic.Name == TGSI_SEMANTIC_POSITION) :
 503                             (d->Semantic.Name == TGSI_SEMANTIC_COLOR)) {
 504                                 if (last_args[0]) {
 505                                         lp_build_intrinsic(base->gallivm->builder,
 506                                                            "llvm.SI.export",
 507                                                            LLVMVoidTypeInContext(base->gallivm->context),
 508                                                            last_args, 9);
 509                                 }
 510
 511                                 memcpy(last_args, args, sizeof(args));
 512                         } else {
 513                                 lp_build_intrinsic(base->gallivm->builder,
 514                                                    "llvm.SI.export",
 515                                                    LLVMVoidTypeInContext(base->gallivm->context),
 516                                                    args, 9);
 517                         }
 518
 519                 }
 520         }
 521
 522         if (!last_args[0]) {
 523                 assert(si_shader_ctx->type == TGSI_PROCESSOR_FRAGMENT);
 524
 525                 /* Specify which components to enable */
 526                 last_args[0] = lp_build_const_int32(base->gallivm, 0x0);
 527
 528                 /* Specify the target we are exporting */
 529                 last_args[3] = lp_build_const_int32(base->gallivm, V_008DFC_SQ_EXP_MRT);
 530
 531                 /* Set COMPR flag to zero to export data as 32-bit */
 532                 last_args[4] = uint->zero;
 533
 534                 /* dummy bits */
 535                 last_args[5]= uint->zero;
 536                 last_args[6]= uint->zero;
 537                 last_args[7]= uint->zero;
 538                 last_args[8]= uint->zero;
 539         }
 540
 541         /* Specify whether the EXEC mask represents the valid mask */
 542         last_args[1] = lp_build_const_int32(base->gallivm,
 543                                             si_shader_ctx->type == TGSI_PROCESSOR_FRAGMENT);
 544
 545         /* Specify that this is the last export */
 546         last_args[2] = lp_build_const_int32(base->gallivm, 1);
 547
 548         lp_build_intrinsic(base->gallivm->builder,
 549                            "llvm.SI.export",
 550                            LLVMVoidTypeInContext(base->gallivm->context),
 551                            last_args, 9);
 552
 553 /* XXX: Look up what this function does */
 554 /*              ctx->shader->output[i].spi_sid = r600_spi_sid(&ctx->shader->output[i]);*/
 555 }
 556
 557 static void tex_fetch_args(
 558         struct lp_build_tgsi_context * bld_base,
 559         struct lp_build_emit_data * emit_data)
 560 {
 561         const struct tgsi_full_instruction * inst = emit_data->inst;
 562         LLVMValueRef ptr;
 563         LLVMValueRef offset;
 564
 565         /* WriteMask */
 566         /* XXX: should be optimized using emit_data->inst->Dst[0].Register.WriteMask*/
 567         emit_data->args[0] = lp_build_const_int32(bld_base->base.gallivm, 0xf);
 568
 569         /* Coordinates */
 570         /* XXX: Not all sample instructions need 4 address arguments. */
 571         if (inst->Instruction.Opcode == TGSI_OPCODE_TXP) {
 572                 LLVMValueRef src_w;
 573                 unsigned chan;
 574                 LLVMValueRef coords[4];
 575
 576                 emit_data->dst_type = LLVMVectorType(bld_base->base.elem_type, 4);
 577                 src_w = lp_build_emit_fetch(bld_base, emit_data->inst, 0, TGSI_CHAN_W);
 578
 579                 for (chan = 0; chan < 3; chan++ ) {
 580                         LLVMValueRef arg = lp_build_emit_fetch(bld_base,
 581                                                                emit_data->inst, 0, chan);
 582                         coords[chan] = lp_build_emit_llvm_binary(bld_base,
 583                                                                  TGSI_OPCODE_DIV,
 584                                                                  arg, src_w);
 585                 }
 586                 coords[3] = bld_base->base.one;
 587                 emit_data->args[1] = lp_build_gather_values(bld_base->base.gallivm,
 588                                                             coords, 4);
 589         } else
 590                 emit_data->args[1] = lp_build_emit_fetch(bld_base, emit_data->inst,
 591                                                          0, LP_CHAN_ALL);
 592
 593         /* Resource */
 594         ptr = use_sgpr(bld_base->base.gallivm, SGPR_CONST_PTR_V8I32, 4);
 595         offset = lp_build_const_int32(bld_base->base.gallivm,
 596                                   emit_data->inst->Src[1].Register.Index);
 597         emit_data->args[2] = build_indexed_load(bld_base->base.gallivm,
 598                                                 ptr, offset);
 599
 600         /* Sampler */
 601         ptr = use_sgpr(bld_base->base.gallivm, SGPR_CONST_PTR_V4I32, 2);
 602         offset = lp_build_const_int32(bld_base->base.gallivm,
 603                                   emit_data->inst->Src[1].Register.Index);
 604         emit_data->args[3] = build_indexed_load(bld_base->base.gallivm,
 605                                                 ptr, offset);
 606
 607         /* Dimensions */
 608         /* XXX: We might want to pass this information to the shader at some. */
 609 /*      emit_data->args[4] = lp_build_const_int32(bld_base->base.gallivm,
 610                                         emit_data->inst->Texture.Texture);
 611 */
 612
 613         emit_data->arg_count = 4;
 614         /* XXX: To optimize, we could use a float or v2f32, if the last bits of
 615          * the writemask are clear */
 616         emit_data->dst_type = LLVMVectorType(
 617                         LLVMFloatTypeInContext(bld_base->base.gallivm->context),
 618                         4);
 619 }
 620
 621 static const struct lp_build_tgsi_action tex_action = {
 622         .fetch_args = tex_fetch_args,
 623         .emit = lp_build_tgsi_intrinsic,
 624         .intr_name = "llvm.SI.sample"
 625 };
 626
 627
 628 int si_pipe_shader_create(
 629         struct pipe_context *ctx,
 630         struct si_pipe_shader *shader)
 631 {
 632         struct r600_context *rctx = (struct r600_context*)ctx;
 633         struct si_pipe_shader_selector *sel = shader->selector;
 634         struct si_shader_context si_shader_ctx;
 635         struct tgsi_shader_info shader_info;
 636         struct lp_build_tgsi_context * bld_base;
 637         LLVMModuleRef mod;
 638         unsigned char * inst_bytes;
 639         unsigned inst_byte_count;
 640         unsigned i;
 641         uint32_t *ptr;
 642         bool dump;
 643
 644         dump = debug_get_bool_option("RADEON_DUMP_SHADERS", FALSE);
 645
 646         memset(&si_shader_ctx.radeon_bld, 0, sizeof(si_shader_ctx.radeon_bld));
 647         radeon_llvm_context_init(&si_shader_ctx.radeon_bld);
 648         bld_base = &si_shader_ctx.radeon_bld.soa.bld_base;
 649
 650         tgsi_scan_shader(sel->tokens, &shader_info);
 651         bld_base->info = &shader_info;
 652         bld_base->emit_fetch_funcs[TGSI_FILE_CONSTANT] = fetch_constant;
 653         bld_base->emit_epilogue = si_llvm_emit_epilogue;
 654
 655         bld_base->op_actions[TGSI_OPCODE_TEX] = tex_action;
 656         bld_base->op_actions[TGSI_OPCODE_TXP] = tex_action;
 657
 658         si_shader_ctx.radeon_bld.load_input = declare_input;
 659         si_shader_ctx.tokens = sel->tokens;
 660         tgsi_parse_init(&si_shader_ctx.parse, si_shader_ctx.tokens);
 661         si_shader_ctx.shader = shader;
 662         si_shader_ctx.type = si_shader_ctx.parse.FullHeader.Processor.Processor;
 663         si_shader_ctx.rctx = rctx;
 664
 665         shader->shader.nr_cbufs = rctx->framebuffer.nr_cbufs;
 666
 667         /* Dump TGSI code before doing TGSI->LLVM conversion in case the
 668          * conversion fails. */
 669         if (dump) {
 670                 tgsi_dump(sel->tokens, 0);
 671         }
 672
 673         if (!lp_build_tgsi_llvm(bld_base, sel->tokens)) {
 674                 fprintf(stderr, "Failed to translate shader from TGSI to LLVM\n");
 675                 return -EINVAL;
 676         }
 677
 678         radeon_llvm_finalize_module(&si_shader_ctx.radeon_bld);
 679
 680         mod = bld_base->base.gallivm->module;
 681         if (dump) {
 682                 LLVMDumpModule(mod);
 683         }
 684         radeon_llvm_compile(mod, &inst_bytes, &inst_byte_count, "SI", dump);
 685         if (dump) {
 686                 fprintf(stderr, "SI CODE:\n");
 687                 for (i = 0; i < inst_byte_count; i+=4 ) {
 688                         fprintf(stderr, "%02x%02x%02x%02x\n", inst_bytes[i + 3],
 689                                 inst_bytes[i + 2], inst_bytes[i + 1],
 690                                 inst_bytes[i]);
 691                 }
 692         }
 693
 694         shader->num_sgprs = util_le32_to_cpu(*(uint32_t*)inst_bytes);
 695         shader->num_vgprs = util_le32_to_cpu(*(uint32_t*)(inst_bytes + 4));
 696         shader->spi_ps_input_ena = util_le32_to_cpu(*(uint32_t*)(inst_bytes + 8));
 697
 698         radeon_llvm_dispose(&si_shader_ctx.radeon_bld);
 699         tgsi_parse_free(&si_shader_ctx.parse);
 700
 701         /* copy new shader */
 702         si_resource_reference(&shader->bo, NULL);
 703         shader->bo = si_resource_create_custom(ctx->screen, PIPE_USAGE_IMMUTABLE,
 704                                                inst_byte_count - 12);
 705         if (shader->bo == NULL) {
 706                 return -ENOMEM;
 707         }
 708
 709         ptr = (uint32_t*)rctx->ws->buffer_map(shader->bo->cs_buf, rctx->cs, PIPE_TRANSFER_WRITE);
 710         if (0 /*R600_BIG_ENDIAN*/) {
 711                 for (i = 0; i < (inst_byte_count-12)/4; ++i) {
 712                         ptr[i] = util_bswap32(*(uint32_t*)(inst_bytes+12 + i*4));
 713                 }
 714         } else {
 715                 memcpy(ptr, inst_bytes + 12, inst_byte_count - 12);
 716         }
 717         rctx->ws->buffer_unmap(shader->bo->cs_buf);
 718
 719         free(inst_bytes);
 720
 721         return 0;
 722 }
 723
 724 void si_pipe_shader_destroy(struct pipe_context *ctx, struct si_pipe_shader *shader)
 725 {
 726         si_resource_reference(&shader->bo, NULL);
 727 }