src/gallium/drivers/radeonsi/si_shaderlib_tgsi.c

   1 /*
   2  * Copyright 2018 Advanced Micro Devices, Inc.
   3  * All Rights Reserved.
   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
  25 #include "si_pipe.h"
  26 #include "tgsi/tgsi_text.h"
  27 #include "tgsi/tgsi_ureg.h"
  28
  29 void *si_get_blitter_vs(struct si_context *sctx, enum blitter_attrib_type type,
  30                         unsigned num_layers)
  31 {
  32         unsigned vs_blit_property;
  33         void **vs;
  34
  35         switch (type) {
  36         case UTIL_BLITTER_ATTRIB_NONE:
  37                 vs = num_layers > 1 ? &sctx->vs_blit_pos_layered :
  38                                       &sctx->vs_blit_pos;
  39                 vs_blit_property = SI_VS_BLIT_SGPRS_POS;
  40                 break;
  41         case UTIL_BLITTER_ATTRIB_COLOR:
  42                 vs = num_layers > 1 ? &sctx->vs_blit_color_layered :
  43                                       &sctx->vs_blit_color;
  44                 vs_blit_property = SI_VS_BLIT_SGPRS_POS_COLOR;
  45                 break;
  46         case UTIL_BLITTER_ATTRIB_TEXCOORD_XY:
  47         case UTIL_BLITTER_ATTRIB_TEXCOORD_XYZW:
  48                 assert(num_layers == 1);
  49                 vs = &sctx->vs_blit_texcoord;
  50                 vs_blit_property = SI_VS_BLIT_SGPRS_POS_TEXCOORD;
  51                 break;
  52         default:
  53                 assert(0);
  54                 return NULL;
  55         }
  56         if (*vs)
  57                 return *vs;
  58
  59         struct ureg_program *ureg = ureg_create(PIPE_SHADER_VERTEX);
  60         if (!ureg)
  61                 return NULL;
  62
  63         /* Tell the shader to load VS inputs from SGPRs: */
  64         ureg_property(ureg, TGSI_PROPERTY_VS_BLIT_SGPRS, vs_blit_property);
  65         ureg_property(ureg, TGSI_PROPERTY_VS_WINDOW_SPACE_POSITION, true);
  66
  67         /* This is just a pass-through shader with 1-3 MOV instructions. */
  68         ureg_MOV(ureg,
  69                  ureg_DECL_output(ureg, TGSI_SEMANTIC_POSITION, 0),
  70                  ureg_DECL_vs_input(ureg, 0));
  71
  72         if (type != UTIL_BLITTER_ATTRIB_NONE) {
  73                 ureg_MOV(ureg,
  74                          ureg_DECL_output(ureg, TGSI_SEMANTIC_GENERIC, 0),
  75                          ureg_DECL_vs_input(ureg, 1));
  76         }
  77
  78         if (num_layers > 1) {
  79                 struct ureg_src instance_id =
  80                         ureg_DECL_system_value(ureg, TGSI_SEMANTIC_INSTANCEID, 0);
  81                 struct ureg_dst layer =
  82                         ureg_DECL_output(ureg, TGSI_SEMANTIC_LAYER, 0);
  83
  84                 ureg_MOV(ureg, ureg_writemask(layer, TGSI_WRITEMASK_X),
  85                          ureg_scalar(instance_id, TGSI_SWIZZLE_X));
  86         }
  87         ureg_END(ureg);
  88
  89         *vs = ureg_create_shader_and_destroy(ureg, &sctx->b);
  90         return *vs;
  91 }
  92
  93 /**
  94  * This is used when TCS is NULL in the VS->TCS->TES chain. In this case,
  95  * VS passes its outputs to TES directly, so the fixed-function shader only
  96  * has to write TESSOUTER and TESSINNER.
  97  */
  98 void *si_create_fixed_func_tcs(struct si_context *sctx)
  99 {
 100         struct ureg_src outer, inner;
 101         struct ureg_dst tessouter, tessinner;
 102         struct ureg_program *ureg = ureg_create(PIPE_SHADER_TESS_CTRL);
 103
 104         if (!ureg)
 105                 return NULL;
 106
 107         outer = ureg_DECL_system_value(ureg,
 108                                        TGSI_SEMANTIC_DEFAULT_TESSOUTER_SI, 0);
 109         inner = ureg_DECL_system_value(ureg,
 110                                        TGSI_SEMANTIC_DEFAULT_TESSINNER_SI, 0);
 111
 112         tessouter = ureg_DECL_output(ureg, TGSI_SEMANTIC_TESSOUTER, 0);
 113         tessinner = ureg_DECL_output(ureg, TGSI_SEMANTIC_TESSINNER, 0);
 114
 115         ureg_MOV(ureg, tessouter, outer);
 116         ureg_MOV(ureg, tessinner, inner);
 117         ureg_END(ureg);
 118
 119         return ureg_create_shader_and_destroy(ureg, &sctx->b);
 120 }
 121
 122 /* Create a compute shader implementing clear_buffer or copy_buffer. */
 123 void *si_create_dma_compute_shader(struct pipe_context *ctx,
 124                                    unsigned num_dwords_per_thread,
 125                                    bool dst_stream_cache_policy, bool is_copy)
 126 {
 127         assert(util_is_power_of_two_nonzero(num_dwords_per_thread));
 128
 129         unsigned store_qualifier = TGSI_MEMORY_COHERENT | TGSI_MEMORY_RESTRICT;
 130         if (dst_stream_cache_policy)
 131                 store_qualifier |= TGSI_MEMORY_STREAM_CACHE_POLICY;
 132
 133         /* Don't cache loads, because there is no reuse. */
 134         unsigned load_qualifier = store_qualifier | TGSI_MEMORY_STREAM_CACHE_POLICY;
 135
 136         unsigned num_mem_ops = MAX2(1, num_dwords_per_thread / 4);
 137         unsigned *inst_dwords = alloca(num_mem_ops * sizeof(unsigned));
 138
 139         for (unsigned i = 0; i < num_mem_ops; i++) {
 140                 if (i*4 < num_dwords_per_thread)
 141                         inst_dwords[i] = MIN2(4, num_dwords_per_thread - i*4);
 142         }
 143
 144         struct ureg_program *ureg = ureg_create(PIPE_SHADER_COMPUTE);
 145         if (!ureg)
 146                 return NULL;
 147
 148         ureg_property(ureg, TGSI_PROPERTY_CS_FIXED_BLOCK_WIDTH, 64);
 149         ureg_property(ureg, TGSI_PROPERTY_CS_FIXED_BLOCK_HEIGHT, 1);
 150         ureg_property(ureg, TGSI_PROPERTY_CS_FIXED_BLOCK_DEPTH, 1);
 151
 152         struct ureg_src value;
 153         if (!is_copy) {
 154                 ureg_property(ureg, TGSI_PROPERTY_CS_USER_DATA_DWORDS, inst_dwords[0]);
 155                 value = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_CS_USER_DATA, 0);
 156         }
 157
 158         struct ureg_src tid = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_THREAD_ID, 0);
 159         struct ureg_src blk = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_BLOCK_ID, 0);
 160         struct ureg_dst store_addr = ureg_writemask(ureg_DECL_temporary(ureg), TGSI_WRITEMASK_X);
 161         struct ureg_dst load_addr = ureg_writemask(ureg_DECL_temporary(ureg), TGSI_WRITEMASK_X);
 162         struct ureg_dst dstbuf = ureg_dst(ureg_DECL_buffer(ureg, 0, false));
 163         struct ureg_src srcbuf;
 164         struct ureg_src *values = NULL;
 165
 166         if (is_copy) {
 167                 srcbuf = ureg_DECL_buffer(ureg, 1, false);
 168                 values = malloc(num_mem_ops * sizeof(struct ureg_src));
 169         }
 170
 171         /* If there are multiple stores, the first store writes into 0+tid,
 172          * the 2nd store writes into 64+tid, the 3rd store writes into 128+tid, etc.
 173          */
 174         ureg_UMAD(ureg, store_addr, blk, ureg_imm1u(ureg, 64 * num_mem_ops), tid);
 175         /* Convert from a "store size unit" into bytes. */
 176         ureg_UMUL(ureg, store_addr, ureg_src(store_addr),
 177                   ureg_imm1u(ureg, 4 * inst_dwords[0]));
 178         ureg_MOV(ureg, load_addr, ureg_src(store_addr));
 179
 180         /* Distance between a load and a store for latency hiding. */
 181         unsigned load_store_distance = is_copy ? 8 : 0;
 182
 183         for (unsigned i = 0; i < num_mem_ops + load_store_distance; i++) {
 184                 int d = i - load_store_distance;
 185
 186                 if (is_copy && i < num_mem_ops) {
 187                         if (i) {
 188                                 ureg_UADD(ureg, load_addr, ureg_src(load_addr),
 189                                           ureg_imm1u(ureg, 4 * inst_dwords[i] * 64));
 190                         }
 191
 192                         values[i] = ureg_src(ureg_DECL_temporary(ureg));
 193                         struct ureg_dst dst =
 194                                 ureg_writemask(ureg_dst(values[i]),
 195                                                u_bit_consecutive(0, inst_dwords[i]));
 196                         struct ureg_src srcs[] = {srcbuf, ureg_src(load_addr)};
 197                         ureg_memory_insn(ureg, TGSI_OPCODE_LOAD, &dst, 1, srcs, 2,
 198                                          load_qualifier, TGSI_TEXTURE_BUFFER, 0);
 199                 }
 200
 201                 if (d >= 0) {
 202                         if (d) {
 203                                 ureg_UADD(ureg, store_addr, ureg_src(store_addr),
 204                                           ureg_imm1u(ureg, 4 * inst_dwords[d] * 64));
 205                         }
 206
 207                         struct ureg_dst dst =
 208                                 ureg_writemask(dstbuf, u_bit_consecutive(0, inst_dwords[d]));
 209                         struct ureg_src srcs[] =
 210                                 {ureg_src(store_addr), is_copy ? values[d] : value};
 211                         ureg_memory_insn(ureg, TGSI_OPCODE_STORE, &dst, 1, srcs, 2,
 212                                          store_qualifier, TGSI_TEXTURE_BUFFER, 0);
 213                 }
 214         }
 215         ureg_END(ureg);
 216
 217         struct pipe_compute_state state = {};
 218         state.ir_type = PIPE_SHADER_IR_TGSI;
 219         state.prog = ureg_get_tokens(ureg, NULL);
 220
 221         void *cs = ctx->create_compute_state(ctx, &state);
 222         ureg_destroy(ureg);
 223         ureg_free_tokens(state.prog);
 224
 225         free(values);
 226         return cs;
 227 }
 228
 229 /* Create the compute shader that is used to collect the results.
 230  *
 231  * One compute grid with a single thread is launched for every query result
 232  * buffer. The thread (optionally) reads a previous summary buffer, then
 233  * accumulates data from the query result buffer, and writes the result either
 234  * to a summary buffer to be consumed by the next grid invocation or to the
 235  * user-supplied buffer.
 236  *
 237  * Data layout:
 238  *
 239  * CONST
 240  *  0.x = end_offset
 241  *  0.y = result_stride
 242  *  0.z = result_count
 243  *  0.w = bit field:
 244  *          1: read previously accumulated values
 245  *          2: write accumulated values for chaining
 246  *          4: write result available
 247  *          8: convert result to boolean (0/1)
 248  *         16: only read one dword and use that as result
 249  *         32: apply timestamp conversion
 250  *         64: store full 64 bits result
 251  *        128: store signed 32 bits result
 252  *        256: SO_OVERFLOW mode: take the difference of two successive half-pairs
 253  *  1.x = fence_offset
 254  *  1.y = pair_stride
 255  *  1.z = pair_count
 256  *
 257  * BUFFER[0] = query result buffer
 258  * BUFFER[1] = previous summary buffer
 259  * BUFFER[2] = next summary buffer or user-supplied buffer
 260  */
 261 void *si_create_query_result_cs(struct si_context *sctx)
 262 {
 263         /* TEMP[0].xy = accumulated result so far
 264          * TEMP[0].z = result not available
 265          *
 266          * TEMP[1].x = current result index
 267          * TEMP[1].y = current pair index
 268          */
 269         static const char text_tmpl[] =
 270                 "COMP\n"
 271                 "PROPERTY CS_FIXED_BLOCK_WIDTH 1\n"
 272                 "PROPERTY CS_FIXED_BLOCK_HEIGHT 1\n"
 273                 "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n"
 274                 "DCL BUFFER[0]\n"
 275                 "DCL BUFFER[1]\n"
 276                 "DCL BUFFER[2]\n"
 277                 "DCL CONST[0][0..1]\n"
 278                 "DCL TEMP[0..5]\n"
 279                 "IMM[0] UINT32 {0, 31, 2147483647, 4294967295}\n"
 280                 "IMM[1] UINT32 {1, 2, 4, 8}\n"
 281                 "IMM[2] UINT32 {16, 32, 64, 128}\n"
 282                 "IMM[3] UINT32 {1000000, 0, %u, 0}\n" /* for timestamp conversion */
 283                 "IMM[4] UINT32 {256, 0, 0, 0}\n"
 284
 285                 "AND TEMP[5], CONST[0][0].wwww, IMM[2].xxxx\n"
 286                 "UIF TEMP[5]\n"
 287                         /* Check result availability. */
 288                         "LOAD TEMP[1].x, BUFFER[0], CONST[0][1].xxxx\n"
 289                         "ISHR TEMP[0].z, TEMP[1].xxxx, IMM[0].yyyy\n"
 290                         "MOV TEMP[1], TEMP[0].zzzz\n"
 291                         "NOT TEMP[0].z, TEMP[0].zzzz\n"
 292
 293                         /* Load result if available. */
 294                         "UIF TEMP[1]\n"
 295                                 "LOAD TEMP[0].xy, BUFFER[0], IMM[0].xxxx\n"
 296                         "ENDIF\n"
 297                 "ELSE\n"
 298                         /* Load previously accumulated result if requested. */
 299                         "MOV TEMP[0], IMM[0].xxxx\n"
 300                         "AND TEMP[4], CONST[0][0].wwww, IMM[1].xxxx\n"
 301                         "UIF TEMP[4]\n"
 302                                 "LOAD TEMP[0].xyz, BUFFER[1], IMM[0].xxxx\n"
 303                         "ENDIF\n"
 304
 305                         "MOV TEMP[1].x, IMM[0].xxxx\n"
 306                         "BGNLOOP\n"
 307                                 /* Break if accumulated result so far is not available. */
 308                                 "UIF TEMP[0].zzzz\n"
 309                                         "BRK\n"
 310                                 "ENDIF\n"
 311
 312                                 /* Break if result_index >= result_count. */
 313                                 "USGE TEMP[5], TEMP[1].xxxx, CONST[0][0].zzzz\n"
 314                                 "UIF TEMP[5]\n"
 315                                         "BRK\n"
 316                                 "ENDIF\n"
 317
 318                                 /* Load fence and check result availability */
 319                                 "UMAD TEMP[5].x, TEMP[1].xxxx, CONST[0][0].yyyy, CONST[0][1].xxxx\n"
 320                                 "LOAD TEMP[5].x, BUFFER[0], TEMP[5].xxxx\n"
 321                                 "ISHR TEMP[0].z, TEMP[5].xxxx, IMM[0].yyyy\n"
 322                                 "NOT TEMP[0].z, TEMP[0].zzzz\n"
 323                                 "UIF TEMP[0].zzzz\n"
 324                                         "BRK\n"
 325                                 "ENDIF\n"
 326
 327                                 "MOV TEMP[1].y, IMM[0].xxxx\n"
 328                                 "BGNLOOP\n"
 329                                         /* Load start and end. */
 330                                         "UMUL TEMP[5].x, TEMP[1].xxxx, CONST[0][0].yyyy\n"
 331                                         "UMAD TEMP[5].x, TEMP[1].yyyy, CONST[0][1].yyyy, TEMP[5].xxxx\n"
 332                                         "LOAD TEMP[2].xy, BUFFER[0], TEMP[5].xxxx\n"
 333
 334                                         "UADD TEMP[5].y, TEMP[5].xxxx, CONST[0][0].xxxx\n"
 335                                         "LOAD TEMP[3].xy, BUFFER[0], TEMP[5].yyyy\n"
 336
 337                                         "U64ADD TEMP[4].xy, TEMP[3], -TEMP[2]\n"
 338
 339                                         "AND TEMP[5].z, CONST[0][0].wwww, IMM[4].xxxx\n"
 340                                         "UIF TEMP[5].zzzz\n"
 341                                                 /* Load second start/end half-pair and
 342                                                  * take the difference
 343                                                  */
 344                                                 "UADD TEMP[5].xy, TEMP[5], IMM[1].wwww\n"
 345                                                 "LOAD TEMP[2].xy, BUFFER[0], TEMP[5].xxxx\n"
 346                                                 "LOAD TEMP[3].xy, BUFFER[0], TEMP[5].yyyy\n"
 347
 348                                                 "U64ADD TEMP[3].xy, TEMP[3], -TEMP[2]\n"
 349                                                 "U64ADD TEMP[4].xy, TEMP[4], -TEMP[3]\n"
 350                                         "ENDIF\n"
 351
 352                                         "U64ADD TEMP[0].xy, TEMP[0], TEMP[4]\n"
 353
 354                                         /* Increment pair index */
 355                                         "UADD TEMP[1].y, TEMP[1].yyyy, IMM[1].xxxx\n"
 356                                         "USGE TEMP[5], TEMP[1].yyyy, CONST[0][1].zzzz\n"
 357                                         "UIF TEMP[5]\n"
 358                                                 "BRK\n"
 359                                         "ENDIF\n"
 360                                 "ENDLOOP\n"
 361
 362                                 /* Increment result index */
 363                                 "UADD TEMP[1].x, TEMP[1].xxxx, IMM[1].xxxx\n"
 364                         "ENDLOOP\n"
 365                 "ENDIF\n"
 366
 367                 "AND TEMP[4], CONST[0][0].wwww, IMM[1].yyyy\n"
 368                 "UIF TEMP[4]\n"
 369                         /* Store accumulated data for chaining. */
 370                         "STORE BUFFER[2].xyz, IMM[0].xxxx, TEMP[0]\n"
 371                 "ELSE\n"
 372                         "AND TEMP[4], CONST[0][0].wwww, IMM[1].zzzz\n"
 373                         "UIF TEMP[4]\n"
 374                                 /* Store result availability. */
 375                                 "NOT TEMP[0].z, TEMP[0]\n"
 376                                 "AND TEMP[0].z, TEMP[0].zzzz, IMM[1].xxxx\n"
 377                                 "STORE BUFFER[2].x, IMM[0].xxxx, TEMP[0].zzzz\n"
 378
 379                                 "AND TEMP[4], CONST[0][0].wwww, IMM[2].zzzz\n"
 380                                 "UIF TEMP[4]\n"
 381                                         "STORE BUFFER[2].y, IMM[0].xxxx, IMM[0].xxxx\n"
 382                                 "ENDIF\n"
 383                         "ELSE\n"
 384                                 /* Store result if it is available. */
 385                                 "NOT TEMP[4], TEMP[0].zzzz\n"
 386                                 "UIF TEMP[4]\n"
 387                                         /* Apply timestamp conversion */
 388                                         "AND TEMP[4], CONST[0][0].wwww, IMM[2].yyyy\n"
 389                                         "UIF TEMP[4]\n"
 390                                                 "U64MUL TEMP[0].xy, TEMP[0], IMM[3].xyxy\n"
 391                                                 "U64DIV TEMP[0].xy, TEMP[0], IMM[3].zwzw\n"
 392                                         "ENDIF\n"
 393
 394                                         /* Convert to boolean */
 395                                         "AND TEMP[4], CONST[0][0].wwww, IMM[1].wwww\n"
 396                                         "UIF TEMP[4]\n"
 397                                                 "U64SNE TEMP[0].x, TEMP[0].xyxy, IMM[4].zwzw\n"
 398                                                 "AND TEMP[0].x, TEMP[0].xxxx, IMM[1].xxxx\n"
 399                                                 "MOV TEMP[0].y, IMM[0].xxxx\n"
 400                                         "ENDIF\n"
 401
 402                                         "AND TEMP[4], CONST[0][0].wwww, IMM[2].zzzz\n"
 403                                         "UIF TEMP[4]\n"
 404                                                 "STORE BUFFER[2].xy, IMM[0].xxxx, TEMP[0].xyxy\n"
 405                                         "ELSE\n"
 406                                                 /* Clamping */
 407                                                 "UIF TEMP[0].yyyy\n"
 408                                                         "MOV TEMP[0].x, IMM[0].wwww\n"
 409                                                 "ENDIF\n"
 410
 411                                                 "AND TEMP[4], CONST[0][0].wwww, IMM[2].wwww\n"
 412                                                 "UIF TEMP[4]\n"
 413                                                         "UMIN TEMP[0].x, TEMP[0].xxxx, IMM[0].zzzz\n"
 414                                                 "ENDIF\n"
 415
 416                                                 "STORE BUFFER[2].x, IMM[0].xxxx, TEMP[0].xxxx\n"
 417                                         "ENDIF\n"
 418                                 "ENDIF\n"
 419                         "ENDIF\n"
 420                 "ENDIF\n"
 421
 422                 "END\n";
 423
 424         char text[sizeof(text_tmpl) + 32];
 425         struct tgsi_token tokens[1024];
 426         struct pipe_compute_state state = {};
 427
 428         /* Hard code the frequency into the shader so that the backend can
 429          * use the full range of optimizations for divide-by-constant.
 430          */
 431         snprintf(text, sizeof(text), text_tmpl,
 432                  sctx->screen->info.clock_crystal_freq);
 433
 434         if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) {
 435                 assert(false);
 436                 return NULL;
 437         }
 438
 439         state.ir_type = PIPE_SHADER_IR_TGSI;
 440         state.prog = tokens;
 441
 442         return sctx->b.create_compute_state(&sctx->b, &state);
 443 }
 444
 445 /* Create a compute shader implementing copy_image.
 446  * Luckily, this works with all texture targets except 1D_ARRAY.
 447  */
 448 void *si_create_copy_image_compute_shader(struct pipe_context *ctx)
 449 {
 450         static const char text[] =
 451                 "COMP\n"
 452                 "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n"
 453                 "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n"
 454                 "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n"
 455                 "DCL SV[0], THREAD_ID\n"
 456                 "DCL SV[1], BLOCK_ID\n"
 457                 "DCL IMAGE[0], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n"
 458                 "DCL IMAGE[1], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n"
 459                 "DCL CONST[0][0..1]\n" // 0:xyzw 1:xyzw
 460                 "DCL TEMP[0..4], LOCAL\n"
 461                 "IMM[0] UINT32 {8, 1, 0, 0}\n"
 462                 "MOV TEMP[0].xyz, CONST[0][0].xyzw\n"
 463                 "UMAD TEMP[1].xyz, SV[1].xyzz, IMM[0].xxyy, SV[0].xyzz\n"
 464                 "UADD TEMP[2].xyz, TEMP[1].xyzx, TEMP[0].xyzx\n"
 465                 "LOAD TEMP[3], IMAGE[0], TEMP[2].xyzx, 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n"
 466                 "MOV TEMP[4].xyz, CONST[0][1].xyzw\n"
 467                 "UADD TEMP[2].xyz, TEMP[1].xyzx, TEMP[4].xyzx\n"
 468                 "STORE IMAGE[1], TEMP[2].xyzz, TEMP[3], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n"
 469                 "END\n";
 470
 471         struct tgsi_token tokens[1024];
 472         struct pipe_compute_state state = {0};
 473
 474         if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) {
 475                 assert(false);
 476                 return NULL;
 477         }
 478
 479         state.ir_type = PIPE_SHADER_IR_TGSI;
 480         state.prog = tokens;
 481
 482         return ctx->create_compute_state(ctx, &state);
 483 }
 484
 485 void *si_create_copy_image_compute_shader_1d_array(struct pipe_context *ctx)
 486 {
 487         static const char text[] =
 488                 "COMP\n"
 489                 "PROPERTY CS_FIXED_BLOCK_WIDTH 64\n"
 490                 "PROPERTY CS_FIXED_BLOCK_HEIGHT 1\n"
 491                 "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n"
 492                 "DCL SV[0], THREAD_ID\n"
 493                 "DCL SV[1], BLOCK_ID\n"
 494                 "DCL IMAGE[0], 1D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n"
 495                 "DCL IMAGE[1], 1D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n"
 496                 "DCL CONST[0][0..1]\n" // 0:xyzw 1:xyzw
 497                 "DCL TEMP[0..4], LOCAL\n"
 498                 "IMM[0] UINT32 {64, 1, 0, 0}\n"
 499                 "MOV TEMP[0].xy, CONST[0][0].xzzw\n"
 500                 "UMAD TEMP[1].xy, SV[1].xyzz, IMM[0].xyyy, SV[0].xyzz\n"
 501                 "UADD TEMP[2].xy, TEMP[1].xyzx, TEMP[0].xyzx\n"
 502                 "LOAD TEMP[3], IMAGE[0], TEMP[2].xyzx, 1D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n"
 503                 "MOV TEMP[4].xy, CONST[0][1].xzzw\n"
 504                 "UADD TEMP[2].xy, TEMP[1].xyzx, TEMP[4].xyzx\n"
 505                 "STORE IMAGE[1], TEMP[2].xyzz, TEMP[3], 1D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n"
 506                 "END\n";
 507
 508         struct tgsi_token tokens[1024];
 509         struct pipe_compute_state state = {0};
 510
 511         if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) {
 512                 assert(false);
 513                 return NULL;
 514         }
 515
 516         state.ir_type = PIPE_SHADER_IR_TGSI;
 517         state.prog = tokens;
 518
 519         return ctx->create_compute_state(ctx, &state);
 520 }
 521
 522 void *si_clear_render_target_shader(struct pipe_context *ctx)
 523 {
 524         static const char text[] =
 525                 "COMP\n"
 526                 "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n"
 527                 "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n"
 528                 "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n"
 529                 "DCL SV[0], THREAD_ID\n"
 530                 "DCL SV[1], BLOCK_ID\n"
 531                 "DCL IMAGE[0], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n"
 532                 "DCL CONST[0][0..1]\n" // 0:xyzw 1:xyzw
 533                 "DCL TEMP[0..3], LOCAL\n"
 534                 "IMM[0] UINT32 {8, 1, 0, 0}\n"
 535                 "MOV TEMP[0].xyz, CONST[0][0].xyzw\n"
 536                 "UMAD TEMP[1].xyz, SV[1].xyzz, IMM[0].xxyy, SV[0].xyzz\n"
 537                 "UADD TEMP[2].xyz, TEMP[1].xyzx, TEMP[0].xyzx\n"
 538                 "MOV TEMP[3].xyzw, CONST[0][1].xyzw\n"
 539                 "STORE IMAGE[0], TEMP[2].xyzz, TEMP[3], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n"
 540                 "END\n";
 541
 542         struct tgsi_token tokens[1024];
 543         struct pipe_compute_state state = {0};
 544
 545         if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) {
 546                 assert(false);
 547                 return NULL;
 548         }
 549
 550         state.ir_type = PIPE_SHADER_IR_TGSI;
 551         state.prog = tokens;
 552
 553         return ctx->create_compute_state(ctx, &state);
 554 }
 555
 556 /* TODO: Didn't really test 1D_ARRAY */
 557 void *si_clear_render_target_shader_1d_array(struct pipe_context *ctx)
 558 {
 559         static const char text[] =
 560                 "COMP\n"
 561                 "PROPERTY CS_FIXED_BLOCK_WIDTH 64\n"
 562                 "PROPERTY CS_FIXED_BLOCK_HEIGHT 1\n"
 563                 "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n"
 564                 "DCL SV[0], THREAD_ID\n"
 565                 "DCL SV[1], BLOCK_ID\n"
 566                 "DCL IMAGE[0], 1D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n"
 567                 "DCL CONST[0][0..1]\n" // 0:xyzw 1:xyzw
 568                 "DCL TEMP[0..3], LOCAL\n"
 569                 "IMM[0] UINT32 {64, 1, 0, 0}\n"
 570                 "MOV TEMP[0].xy, CONST[0][0].xzzw\n"
 571                 "UMAD TEMP[1].xy, SV[1].xyzz, IMM[0].xyyy, SV[0].xyzz\n"
 572                 "UADD TEMP[2].xy, TEMP[1].xyzx, TEMP[0].xyzx\n"
 573                 "MOV TEMP[3].xyzw, CONST[0][1].xyzw\n"
 574                 "STORE IMAGE[0], TEMP[2].xyzz, TEMP[3], 1D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n"
 575                 "END\n";
 576
 577         struct tgsi_token tokens[1024];
 578         struct pipe_compute_state state = {0};
 579
 580         if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) {
 581                 assert(false);
 582                 return NULL;
 583         }
 584
 585         state.ir_type = PIPE_SHADER_IR_TGSI;
 586         state.prog = tokens;
 587
 588         return ctx->create_compute_state(ctx, &state);
 589 }