/* * Copyright 2018 Advanced Micro Devices, Inc. * All Rights Reserved. * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associated documentation files (the "Software"), * to deal in the Software without restriction, including without limitation * on the rights to use, copy, modify, merge, publish, distribute, sub * license, and/or sell copies of the Software, and to permit persons to whom * the Software is furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice (including the next * paragraph) shall be included in all copies or substantial portions of the * Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM, * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE * USE OR OTHER DEALINGS IN THE SOFTWARE. */ #include "si_pipe.h" #include "tgsi/tgsi_text.h" #include "tgsi/tgsi_ureg.h" void *si_get_blitter_vs(struct si_context *sctx, enum blitter_attrib_type type, unsigned num_layers) { unsigned vs_blit_property; void **vs; switch (type) { case UTIL_BLITTER_ATTRIB_NONE: vs = num_layers > 1 ? &sctx->vs_blit_pos_layered : &sctx->vs_blit_pos; vs_blit_property = SI_VS_BLIT_SGPRS_POS; break; case UTIL_BLITTER_ATTRIB_COLOR: vs = num_layers > 1 ? &sctx->vs_blit_color_layered : &sctx->vs_blit_color; vs_blit_property = SI_VS_BLIT_SGPRS_POS_COLOR; break; case UTIL_BLITTER_ATTRIB_TEXCOORD_XY: case UTIL_BLITTER_ATTRIB_TEXCOORD_XYZW: assert(num_layers == 1); vs = &sctx->vs_blit_texcoord; vs_blit_property = SI_VS_BLIT_SGPRS_POS_TEXCOORD; break; default: assert(0); return NULL; } if (*vs) return *vs; struct ureg_program *ureg = ureg_create(PIPE_SHADER_VERTEX); if (!ureg) return NULL; /* Tell the shader to load VS inputs from SGPRs: */ ureg_property(ureg, TGSI_PROPERTY_VS_BLIT_SGPRS_AMD, vs_blit_property); ureg_property(ureg, TGSI_PROPERTY_VS_WINDOW_SPACE_POSITION, true); /* This is just a pass-through shader with 1-3 MOV instructions. */ ureg_MOV(ureg, ureg_DECL_output(ureg, TGSI_SEMANTIC_POSITION, 0), ureg_DECL_vs_input(ureg, 0)); if (type != UTIL_BLITTER_ATTRIB_NONE) { ureg_MOV(ureg, ureg_DECL_output(ureg, TGSI_SEMANTIC_GENERIC, 0), ureg_DECL_vs_input(ureg, 1)); } if (num_layers > 1) { struct ureg_src instance_id = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_INSTANCEID, 0); struct ureg_dst layer = ureg_DECL_output(ureg, TGSI_SEMANTIC_LAYER, 0); ureg_MOV(ureg, ureg_writemask(layer, TGSI_WRITEMASK_X), ureg_scalar(instance_id, TGSI_SWIZZLE_X)); } ureg_END(ureg); *vs = ureg_create_shader_and_destroy(ureg, &sctx->b); return *vs; } /** * This is used when TCS is NULL in the VS->TCS->TES chain. In this case, * VS passes its outputs to TES directly, so the fixed-function shader only * has to write TESSOUTER and TESSINNER. */ void *si_create_fixed_func_tcs(struct si_context *sctx) { struct ureg_src outer, inner; struct ureg_dst tessouter, tessinner; struct ureg_program *ureg = ureg_create(PIPE_SHADER_TESS_CTRL); if (!ureg) return NULL; outer = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_TESS_DEFAULT_OUTER_LEVEL, 0); inner = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_TESS_DEFAULT_INNER_LEVEL, 0); tessouter = ureg_DECL_output(ureg, TGSI_SEMANTIC_TESSOUTER, 0); tessinner = ureg_DECL_output(ureg, TGSI_SEMANTIC_TESSINNER, 0); ureg_MOV(ureg, tessouter, outer); ureg_MOV(ureg, tessinner, inner); ureg_END(ureg); return ureg_create_shader_and_destroy(ureg, &sctx->b); } /* Create a compute shader implementing clear_buffer or copy_buffer. */ void *si_create_dma_compute_shader(struct pipe_context *ctx, unsigned num_dwords_per_thread, bool dst_stream_cache_policy, bool is_copy) { struct si_screen *sscreen = (struct si_screen *)ctx->screen; assert(util_is_power_of_two_nonzero(num_dwords_per_thread)); unsigned store_qualifier = TGSI_MEMORY_COHERENT | TGSI_MEMORY_RESTRICT; if (dst_stream_cache_policy) store_qualifier |= TGSI_MEMORY_STREAM_CACHE_POLICY; /* Don't cache loads, because there is no reuse. */ unsigned load_qualifier = store_qualifier | TGSI_MEMORY_STREAM_CACHE_POLICY; unsigned num_mem_ops = MAX2(1, num_dwords_per_thread / 4); unsigned *inst_dwords = alloca(num_mem_ops * sizeof(unsigned)); for (unsigned i = 0; i < num_mem_ops; i++) { if (i * 4 < num_dwords_per_thread) inst_dwords[i] = MIN2(4, num_dwords_per_thread - i * 4); } struct ureg_program *ureg = ureg_create(PIPE_SHADER_COMPUTE); if (!ureg) return NULL; ureg_property(ureg, TGSI_PROPERTY_CS_FIXED_BLOCK_WIDTH, sscreen->compute_wave_size); ureg_property(ureg, TGSI_PROPERTY_CS_FIXED_BLOCK_HEIGHT, 1); ureg_property(ureg, TGSI_PROPERTY_CS_FIXED_BLOCK_DEPTH, 1); struct ureg_src value; if (!is_copy) { ureg_property(ureg, TGSI_PROPERTY_CS_USER_DATA_COMPONENTS_AMD, inst_dwords[0]); value = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_CS_USER_DATA_AMD, 0); } struct ureg_src tid = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_THREAD_ID, 0); struct ureg_src blk = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_BLOCK_ID, 0); struct ureg_dst store_addr = ureg_writemask(ureg_DECL_temporary(ureg), TGSI_WRITEMASK_X); struct ureg_dst load_addr = ureg_writemask(ureg_DECL_temporary(ureg), TGSI_WRITEMASK_X); struct ureg_dst dstbuf = ureg_dst(ureg_DECL_buffer(ureg, 0, false)); struct ureg_src srcbuf; struct ureg_src *values = NULL; if (is_copy) { srcbuf = ureg_DECL_buffer(ureg, 1, false); values = malloc(num_mem_ops * sizeof(struct ureg_src)); } /* If there are multiple stores, the first store writes into 0*wavesize+tid, * the 2nd store writes into 1*wavesize+tid, the 3rd store writes into 2*wavesize+tid, etc. */ ureg_UMAD(ureg, store_addr, blk, ureg_imm1u(ureg, sscreen->compute_wave_size * num_mem_ops), tid); /* Convert from a "store size unit" into bytes. */ ureg_UMUL(ureg, store_addr, ureg_src(store_addr), ureg_imm1u(ureg, 4 * inst_dwords[0])); ureg_MOV(ureg, load_addr, ureg_src(store_addr)); /* Distance between a load and a store for latency hiding. */ unsigned load_store_distance = is_copy ? 8 : 0; for (unsigned i = 0; i < num_mem_ops + load_store_distance; i++) { int d = i - load_store_distance; if (is_copy && i < num_mem_ops) { if (i) { ureg_UADD(ureg, load_addr, ureg_src(load_addr), ureg_imm1u(ureg, 4 * inst_dwords[i] * sscreen->compute_wave_size)); } values[i] = ureg_src(ureg_DECL_temporary(ureg)); struct ureg_dst dst = ureg_writemask(ureg_dst(values[i]), u_bit_consecutive(0, inst_dwords[i])); struct ureg_src srcs[] = {srcbuf, ureg_src(load_addr)}; ureg_memory_insn(ureg, TGSI_OPCODE_LOAD, &dst, 1, srcs, 2, load_qualifier, TGSI_TEXTURE_BUFFER, 0); } if (d >= 0) { if (d) { ureg_UADD(ureg, store_addr, ureg_src(store_addr), ureg_imm1u(ureg, 4 * inst_dwords[d] * sscreen->compute_wave_size)); } struct ureg_dst dst = ureg_writemask(dstbuf, u_bit_consecutive(0, inst_dwords[d])); struct ureg_src srcs[] = {ureg_src(store_addr), is_copy ? values[d] : value}; ureg_memory_insn(ureg, TGSI_OPCODE_STORE, &dst, 1, srcs, 2, store_qualifier, TGSI_TEXTURE_BUFFER, 0); } } ureg_END(ureg); struct pipe_compute_state state = {}; state.ir_type = PIPE_SHADER_IR_TGSI; state.prog = ureg_get_tokens(ureg, NULL); void *cs = ctx->create_compute_state(ctx, &state); ureg_destroy(ureg); ureg_free_tokens(state.prog); free(values); return cs; } /* Create a compute shader that copies DCC from one buffer to another * where each DCC buffer has a different layout. * * image[0]: offset remap table (pairs of ), * 2 pairs are read * image[1]: DCC source buffer, typed r8_uint * image[2]: DCC destination buffer, typed r8_uint */ void *si_create_dcc_retile_cs(struct pipe_context *ctx) { struct ureg_program *ureg = ureg_create(PIPE_SHADER_COMPUTE); if (!ureg) return NULL; ureg_property(ureg, TGSI_PROPERTY_CS_FIXED_BLOCK_WIDTH, 64); ureg_property(ureg, TGSI_PROPERTY_CS_FIXED_BLOCK_HEIGHT, 1); ureg_property(ureg, TGSI_PROPERTY_CS_FIXED_BLOCK_DEPTH, 1); /* Compute the global thread ID (in idx). */ struct ureg_src tid = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_THREAD_ID, 0); struct ureg_src blk = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_BLOCK_ID, 0); struct ureg_dst idx = ureg_writemask(ureg_DECL_temporary(ureg), TGSI_WRITEMASK_X); ureg_UMAD(ureg, idx, blk, ureg_imm1u(ureg, 64), tid); /* Load 2 pairs of offsets for DCC load & store. */ struct ureg_src map = ureg_DECL_image(ureg, 0, TGSI_TEXTURE_BUFFER, 0, false, false); struct ureg_dst offsets = ureg_DECL_temporary(ureg); struct ureg_src map_load_args[] = {map, ureg_src(idx)}; ureg_memory_insn(ureg, TGSI_OPCODE_LOAD, &offsets, 1, map_load_args, 2, TGSI_MEMORY_RESTRICT, TGSI_TEXTURE_BUFFER, 0); struct ureg_src dcc_src = ureg_DECL_image(ureg, 1, TGSI_TEXTURE_BUFFER, 0, false, false); struct ureg_dst dcc_dst = ureg_dst(ureg_DECL_image(ureg, 2, TGSI_TEXTURE_BUFFER, 0, true, false)); struct ureg_dst dcc_value[2]; /* Copy DCC values: * dst[offsets.y] = src[offsets.x]; * dst[offsets.w] = src[offsets.z]; */ for (unsigned i = 0; i < 2; i++) { dcc_value[i] = ureg_writemask(ureg_DECL_temporary(ureg), TGSI_WRITEMASK_X); struct ureg_src load_args[] = {dcc_src, ureg_scalar(ureg_src(offsets), TGSI_SWIZZLE_X + i * 2)}; ureg_memory_insn(ureg, TGSI_OPCODE_LOAD, &dcc_value[i], 1, load_args, 2, TGSI_MEMORY_RESTRICT, TGSI_TEXTURE_BUFFER, 0); } dcc_dst = ureg_writemask(dcc_dst, TGSI_WRITEMASK_X); for (unsigned i = 0; i < 2; i++) { struct ureg_src store_args[] = {ureg_scalar(ureg_src(offsets), TGSI_SWIZZLE_Y + i * 2), ureg_src(dcc_value[i])}; ureg_memory_insn(ureg, TGSI_OPCODE_STORE, &dcc_dst, 1, store_args, 2, TGSI_MEMORY_RESTRICT, TGSI_TEXTURE_BUFFER, 0); } ureg_END(ureg); struct pipe_compute_state state = {}; state.ir_type = PIPE_SHADER_IR_TGSI; state.prog = ureg_get_tokens(ureg, NULL); void *cs = ctx->create_compute_state(ctx, &state); ureg_destroy(ureg); return cs; } /* Create the compute shader that is used to collect the results. * * One compute grid with a single thread is launched for every query result * buffer. The thread (optionally) reads a previous summary buffer, then * accumulates data from the query result buffer, and writes the result either * to a summary buffer to be consumed by the next grid invocation or to the * user-supplied buffer. * * Data layout: * * CONST * 0.x = end_offset * 0.y = result_stride * 0.z = result_count * 0.w = bit field: * 1: read previously accumulated values * 2: write accumulated values for chaining * 4: write result available * 8: convert result to boolean (0/1) * 16: only read one dword and use that as result * 32: apply timestamp conversion * 64: store full 64 bits result * 128: store signed 32 bits result * 256: SO_OVERFLOW mode: take the difference of two successive half-pairs * 1.x = fence_offset * 1.y = pair_stride * 1.z = pair_count * * BUFFER[0] = query result buffer * BUFFER[1] = previous summary buffer * BUFFER[2] = next summary buffer or user-supplied buffer */ void *si_create_query_result_cs(struct si_context *sctx) { /* TEMP[0].xy = accumulated result so far * TEMP[0].z = result not available * * TEMP[1].x = current result index * TEMP[1].y = current pair index */ static const char text_tmpl[] = "COMP\n" "PROPERTY CS_FIXED_BLOCK_WIDTH 1\n" "PROPERTY CS_FIXED_BLOCK_HEIGHT 1\n" "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" "DCL BUFFER[0]\n" "DCL BUFFER[1]\n" "DCL BUFFER[2]\n" "DCL CONST[0][0..1]\n" "DCL TEMP[0..5]\n" "IMM[0] UINT32 {0, 31, 2147483647, 4294967295}\n" "IMM[1] UINT32 {1, 2, 4, 8}\n" "IMM[2] UINT32 {16, 32, 64, 128}\n" "IMM[3] UINT32 {1000000, 0, %u, 0}\n" /* for timestamp conversion */ "IMM[4] UINT32 {256, 0, 0, 0}\n" "AND TEMP[5], CONST[0][0].wwww, IMM[2].xxxx\n" "UIF TEMP[5]\n" /* Check result availability. */ "LOAD TEMP[1].x, BUFFER[0], CONST[0][1].xxxx\n" "ISHR TEMP[0].z, TEMP[1].xxxx, IMM[0].yyyy\n" "MOV TEMP[1], TEMP[0].zzzz\n" "NOT TEMP[0].z, TEMP[0].zzzz\n" /* Load result if available. */ "UIF TEMP[1]\n" "LOAD TEMP[0].xy, BUFFER[0], IMM[0].xxxx\n" "ENDIF\n" "ELSE\n" /* Load previously accumulated result if requested. */ "MOV TEMP[0], IMM[0].xxxx\n" "AND TEMP[4], CONST[0][0].wwww, IMM[1].xxxx\n" "UIF TEMP[4]\n" "LOAD TEMP[0].xyz, BUFFER[1], IMM[0].xxxx\n" "ENDIF\n" "MOV TEMP[1].x, IMM[0].xxxx\n" "BGNLOOP\n" /* Break if accumulated result so far is not available. */ "UIF TEMP[0].zzzz\n" "BRK\n" "ENDIF\n" /* Break if result_index >= result_count. */ "USGE TEMP[5], TEMP[1].xxxx, CONST[0][0].zzzz\n" "UIF TEMP[5]\n" "BRK\n" "ENDIF\n" /* Load fence and check result availability */ "UMAD TEMP[5].x, TEMP[1].xxxx, CONST[0][0].yyyy, CONST[0][1].xxxx\n" "LOAD TEMP[5].x, BUFFER[0], TEMP[5].xxxx\n" "ISHR TEMP[0].z, TEMP[5].xxxx, IMM[0].yyyy\n" "NOT TEMP[0].z, TEMP[0].zzzz\n" "UIF TEMP[0].zzzz\n" "BRK\n" "ENDIF\n" "MOV TEMP[1].y, IMM[0].xxxx\n" "BGNLOOP\n" /* Load start and end. */ "UMUL TEMP[5].x, TEMP[1].xxxx, CONST[0][0].yyyy\n" "UMAD TEMP[5].x, TEMP[1].yyyy, CONST[0][1].yyyy, TEMP[5].xxxx\n" "LOAD TEMP[2].xy, BUFFER[0], TEMP[5].xxxx\n" "UADD TEMP[5].y, TEMP[5].xxxx, CONST[0][0].xxxx\n" "LOAD TEMP[3].xy, BUFFER[0], TEMP[5].yyyy\n" "U64ADD TEMP[4].xy, TEMP[3], -TEMP[2]\n" "AND TEMP[5].z, CONST[0][0].wwww, IMM[4].xxxx\n" "UIF TEMP[5].zzzz\n" /* Load second start/end half-pair and * take the difference */ "UADD TEMP[5].xy, TEMP[5], IMM[1].wwww\n" "LOAD TEMP[2].xy, BUFFER[0], TEMP[5].xxxx\n" "LOAD TEMP[3].xy, BUFFER[0], TEMP[5].yyyy\n" "U64ADD TEMP[3].xy, TEMP[3], -TEMP[2]\n" "U64ADD TEMP[4].xy, TEMP[4], -TEMP[3]\n" "ENDIF\n" "U64ADD TEMP[0].xy, TEMP[0], TEMP[4]\n" /* Increment pair index */ "UADD TEMP[1].y, TEMP[1].yyyy, IMM[1].xxxx\n" "USGE TEMP[5], TEMP[1].yyyy, CONST[0][1].zzzz\n" "UIF TEMP[5]\n" "BRK\n" "ENDIF\n" "ENDLOOP\n" /* Increment result index */ "UADD TEMP[1].x, TEMP[1].xxxx, IMM[1].xxxx\n" "ENDLOOP\n" "ENDIF\n" "AND TEMP[4], CONST[0][0].wwww, IMM[1].yyyy\n" "UIF TEMP[4]\n" /* Store accumulated data for chaining. */ "STORE BUFFER[2].xyz, IMM[0].xxxx, TEMP[0]\n" "ELSE\n" "AND TEMP[4], CONST[0][0].wwww, IMM[1].zzzz\n" "UIF TEMP[4]\n" /* Store result availability. */ "NOT TEMP[0].z, TEMP[0]\n" "AND TEMP[0].z, TEMP[0].zzzz, IMM[1].xxxx\n" "STORE BUFFER[2].x, IMM[0].xxxx, TEMP[0].zzzz\n" "AND TEMP[4], CONST[0][0].wwww, IMM[2].zzzz\n" "UIF TEMP[4]\n" "STORE BUFFER[2].y, IMM[0].xxxx, IMM[0].xxxx\n" "ENDIF\n" "ELSE\n" /* Store result if it is available. */ "NOT TEMP[4], TEMP[0].zzzz\n" "UIF TEMP[4]\n" /* Apply timestamp conversion */ "AND TEMP[4], CONST[0][0].wwww, IMM[2].yyyy\n" "UIF TEMP[4]\n" "U64MUL TEMP[0].xy, TEMP[0], IMM[3].xyxy\n" "U64DIV TEMP[0].xy, TEMP[0], IMM[3].zwzw\n" "ENDIF\n" /* Convert to boolean */ "AND TEMP[4], CONST[0][0].wwww, IMM[1].wwww\n" "UIF TEMP[4]\n" "U64SNE TEMP[0].x, TEMP[0].xyxy, IMM[4].zwzw\n" "AND TEMP[0].x, TEMP[0].xxxx, IMM[1].xxxx\n" "MOV TEMP[0].y, IMM[0].xxxx\n" "ENDIF\n" "AND TEMP[4], CONST[0][0].wwww, IMM[2].zzzz\n" "UIF TEMP[4]\n" "STORE BUFFER[2].xy, IMM[0].xxxx, TEMP[0].xyxy\n" "ELSE\n" /* Clamping */ "UIF TEMP[0].yyyy\n" "MOV TEMP[0].x, IMM[0].wwww\n" "ENDIF\n" "AND TEMP[4], CONST[0][0].wwww, IMM[2].wwww\n" "UIF TEMP[4]\n" "UMIN TEMP[0].x, TEMP[0].xxxx, IMM[0].zzzz\n" "ENDIF\n" "STORE BUFFER[2].x, IMM[0].xxxx, TEMP[0].xxxx\n" "ENDIF\n" "ENDIF\n" "ENDIF\n" "ENDIF\n" "END\n"; char text[sizeof(text_tmpl) + 32]; struct tgsi_token tokens[1024]; struct pipe_compute_state state = {}; /* Hard code the frequency into the shader so that the backend can * use the full range of optimizations for divide-by-constant. */ snprintf(text, sizeof(text), text_tmpl, sctx->screen->info.clock_crystal_freq); if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) { assert(false); return NULL; } state.ir_type = PIPE_SHADER_IR_TGSI; state.prog = tokens; return sctx->b.create_compute_state(&sctx->b, &state); } /* Create a compute shader implementing copy_image. * Luckily, this works with all texture targets except 1D_ARRAY. */ void *si_create_copy_image_compute_shader(struct pipe_context *ctx) { static const char text[] = "COMP\n" "DCL SV[0], THREAD_ID\n" "DCL SV[1], BLOCK_ID\n" "DCL SV[2], BLOCK_SIZE\n" "DCL IMAGE[0], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n" "DCL IMAGE[1], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n" "DCL CONST[0][0..1]\n" // 0:xyzw 1:xyzw "DCL TEMP[0..4], LOCAL\n" "MOV TEMP[0].xyz, CONST[0][0].xyzw\n" "UMAD TEMP[1].xyz, SV[1].xyzz, SV[2].xyzz, SV[0].xyzz\n" "UADD TEMP[2].xyz, TEMP[1].xyzx, TEMP[0].xyzx\n" "LOAD TEMP[3], IMAGE[0], TEMP[2].xyzx, 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n" "MOV TEMP[4].xyz, CONST[0][1].xyzw\n" "UADD TEMP[2].xyz, TEMP[1].xyzx, TEMP[4].xyzx\n" "STORE IMAGE[1], TEMP[2].xyzz, TEMP[3], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n" "END\n"; struct tgsi_token tokens[1024]; struct pipe_compute_state state = {0}; if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) { assert(false); return NULL; } state.ir_type = PIPE_SHADER_IR_TGSI; state.prog = tokens; return ctx->create_compute_state(ctx, &state); } void *si_create_copy_image_compute_shader_1d_array(struct pipe_context *ctx) { static const char text[] = "COMP\n" "PROPERTY CS_FIXED_BLOCK_WIDTH 64\n" "PROPERTY CS_FIXED_BLOCK_HEIGHT 1\n" "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" "DCL SV[0], THREAD_ID\n" "DCL SV[1], BLOCK_ID\n" "DCL IMAGE[0], 1D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n" "DCL IMAGE[1], 1D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n" "DCL CONST[0][0..1]\n" // 0:xyzw 1:xyzw "DCL TEMP[0..4], LOCAL\n" "IMM[0] UINT32 {64, 1, 0, 0}\n" "MOV TEMP[0].xy, CONST[0][0].xzzw\n" "UMAD TEMP[1].xy, SV[1].xyzz, IMM[0].xyyy, SV[0].xyzz\n" "UADD TEMP[2].xy, TEMP[1].xyzx, TEMP[0].xyzx\n" "LOAD TEMP[3], IMAGE[0], TEMP[2].xyzx, 1D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n" "MOV TEMP[4].xy, CONST[0][1].xzzw\n" "UADD TEMP[2].xy, TEMP[1].xyzx, TEMP[4].xyzx\n" "STORE IMAGE[1], TEMP[2].xyzz, TEMP[3], 1D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n" "END\n"; struct tgsi_token tokens[1024]; struct pipe_compute_state state = {0}; if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) { assert(false); return NULL; } state.ir_type = PIPE_SHADER_IR_TGSI; state.prog = tokens; return ctx->create_compute_state(ctx, &state); } /* Create a compute shader implementing DCC decompression via a blit. * This is a trivial copy_image shader except that it has a variable block * size and a barrier. */ void *si_create_dcc_decompress_cs(struct pipe_context *ctx) { static const char text[] = "COMP\n" "DCL SV[0], THREAD_ID\n" "DCL SV[1], BLOCK_ID\n" "DCL SV[2], BLOCK_SIZE\n" "DCL IMAGE[0], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n" "DCL IMAGE[1], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n" "DCL TEMP[0..1]\n" "UMAD TEMP[0].xyz, SV[1].xyzz, SV[2].xyzz, SV[0].xyzz\n" "LOAD TEMP[1], IMAGE[0], TEMP[0].xyzz, 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n" /* Wait for the whole threadgroup (= DCC block) to load texels before * overwriting them, because overwriting any pixel within a DCC block * can break compression for the whole block. */ "BARRIER\n" "STORE IMAGE[1], TEMP[0].xyzz, TEMP[1], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n" "END\n"; struct tgsi_token tokens[1024]; struct pipe_compute_state state = {0}; if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) { assert(false); return NULL; } state.ir_type = PIPE_SHADER_IR_TGSI; state.prog = tokens; return ctx->create_compute_state(ctx, &state); } void *si_clear_render_target_shader(struct pipe_context *ctx) { static const char text[] = "COMP\n" "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n" "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n" "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" "DCL SV[0], THREAD_ID\n" "DCL SV[1], BLOCK_ID\n" "DCL IMAGE[0], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n" "DCL CONST[0][0..1]\n" // 0:xyzw 1:xyzw "DCL TEMP[0..3], LOCAL\n" "IMM[0] UINT32 {8, 1, 0, 0}\n" "MOV TEMP[0].xyz, CONST[0][0].xyzw\n" "UMAD TEMP[1].xyz, SV[1].xyzz, IMM[0].xxyy, SV[0].xyzz\n" "UADD TEMP[2].xyz, TEMP[1].xyzx, TEMP[0].xyzx\n" "MOV TEMP[3].xyzw, CONST[0][1].xyzw\n" "STORE IMAGE[0], TEMP[2].xyzz, TEMP[3], 2D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n" "END\n"; struct tgsi_token tokens[1024]; struct pipe_compute_state state = {0}; if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) { assert(false); return NULL; } state.ir_type = PIPE_SHADER_IR_TGSI; state.prog = tokens; return ctx->create_compute_state(ctx, &state); } /* TODO: Didn't really test 1D_ARRAY */ void *si_clear_render_target_shader_1d_array(struct pipe_context *ctx) { static const char text[] = "COMP\n" "PROPERTY CS_FIXED_BLOCK_WIDTH 64\n" "PROPERTY CS_FIXED_BLOCK_HEIGHT 1\n" "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" "DCL SV[0], THREAD_ID\n" "DCL SV[1], BLOCK_ID\n" "DCL IMAGE[0], 1D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT, WR\n" "DCL CONST[0][0..1]\n" // 0:xyzw 1:xyzw "DCL TEMP[0..3], LOCAL\n" "IMM[0] UINT32 {64, 1, 0, 0}\n" "MOV TEMP[0].xy, CONST[0][0].xzzw\n" "UMAD TEMP[1].xy, SV[1].xyzz, IMM[0].xyyy, SV[0].xyzz\n" "UADD TEMP[2].xy, TEMP[1].xyzx, TEMP[0].xyzx\n" "MOV TEMP[3].xyzw, CONST[0][1].xyzw\n" "STORE IMAGE[0], TEMP[2].xyzz, TEMP[3], 1D_ARRAY, PIPE_FORMAT_R32G32B32A32_FLOAT\n" "END\n"; struct tgsi_token tokens[1024]; struct pipe_compute_state state = {0}; if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) { assert(false); return NULL; } state.ir_type = PIPE_SHADER_IR_TGSI; state.prog = tokens; return ctx->create_compute_state(ctx, &state); } void *si_clear_12bytes_buffer_shader(struct pipe_context *ctx) { static const char text[] = "COMP\n" "PROPERTY CS_FIXED_BLOCK_WIDTH 64\n" "PROPERTY CS_FIXED_BLOCK_HEIGHT 1\n" "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" "DCL SV[0], THREAD_ID\n" "DCL SV[1], BLOCK_ID\n" "DCL BUFFER[0]\n" "DCL CONST[0][0..0]\n" // 0:xyzw "DCL TEMP[0..0]\n" "IMM[0] UINT32 {64, 1, 12, 0}\n" "UMAD TEMP[0].x, SV[1].xyzz, IMM[0].xyyy, SV[0].xyzz\n" "UMUL TEMP[0].x, TEMP[0].xyzz, IMM[0].zzzz\n" // 12 bytes "STORE BUFFER[0].xyz, TEMP[0].xxxx, CONST[0][0].xyzw\n" "END\n"; struct tgsi_token tokens[1024]; struct pipe_compute_state state = {0}; if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) { assert(false); return NULL; } state.ir_type = PIPE_SHADER_IR_TGSI; state.prog = tokens; return ctx->create_compute_state(ctx, &state); } /* Load samples from the image, and copy them to the same image. This looks like * a no-op, but it's not. Loads use FMASK, while stores don't, so samples are * reordered to match expanded FMASK. * * After the shader finishes, FMASK should be cleared to identity. */ void *si_create_fmask_expand_cs(struct pipe_context *ctx, unsigned num_samples, bool is_array) { enum tgsi_texture_type target = is_array ? TGSI_TEXTURE_2D_ARRAY_MSAA : TGSI_TEXTURE_2D_MSAA; struct ureg_program *ureg = ureg_create(PIPE_SHADER_COMPUTE); if (!ureg) return NULL; ureg_property(ureg, TGSI_PROPERTY_CS_FIXED_BLOCK_WIDTH, 8); ureg_property(ureg, TGSI_PROPERTY_CS_FIXED_BLOCK_HEIGHT, 8); ureg_property(ureg, TGSI_PROPERTY_CS_FIXED_BLOCK_DEPTH, 1); /* Compute the image coordinates. */ struct ureg_src image = ureg_DECL_image(ureg, 0, target, 0, true, false); struct ureg_src tid = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_THREAD_ID, 0); struct ureg_src blk = ureg_DECL_system_value(ureg, TGSI_SEMANTIC_BLOCK_ID, 0); struct ureg_dst coord = ureg_writemask(ureg_DECL_temporary(ureg), TGSI_WRITEMASK_XYZW); ureg_UMAD(ureg, ureg_writemask(coord, TGSI_WRITEMASK_XY), ureg_swizzle(blk, 0, 1, 1, 1), ureg_imm2u(ureg, 8, 8), ureg_swizzle(tid, 0, 1, 1, 1)); if (is_array) { ureg_MOV(ureg, ureg_writemask(coord, TGSI_WRITEMASK_Z), ureg_scalar(blk, TGSI_SWIZZLE_Z)); } /* Load samples, resolving FMASK. */ struct ureg_dst sample[8]; assert(num_samples <= ARRAY_SIZE(sample)); for (unsigned i = 0; i < num_samples; i++) { sample[i] = ureg_DECL_temporary(ureg); ureg_MOV(ureg, ureg_writemask(coord, TGSI_WRITEMASK_W), ureg_imm1u(ureg, i)); struct ureg_src srcs[] = {image, ureg_src(coord)}; ureg_memory_insn(ureg, TGSI_OPCODE_LOAD, &sample[i], 1, srcs, 2, TGSI_MEMORY_RESTRICT, target, 0); } /* Store samples, ignoring FMASK. */ for (unsigned i = 0; i < num_samples; i++) { ureg_MOV(ureg, ureg_writemask(coord, TGSI_WRITEMASK_W), ureg_imm1u(ureg, i)); struct ureg_dst dst_image = ureg_dst(image); struct ureg_src srcs[] = {ureg_src(coord), ureg_src(sample[i])}; ureg_memory_insn(ureg, TGSI_OPCODE_STORE, &dst_image, 1, srcs, 2, TGSI_MEMORY_RESTRICT, target, 0); } ureg_END(ureg); struct pipe_compute_state state = {}; state.ir_type = PIPE_SHADER_IR_TGSI; state.prog = ureg_get_tokens(ureg, NULL); void *cs = ctx->create_compute_state(ctx, &state); ureg_destroy(ureg); return cs; } /* Create the compute shader that is used to collect the results of gfx10+ * shader queries. * * One compute grid with a single thread is launched for every query result * buffer. The thread (optionally) reads a previous summary buffer, then * accumulates data from the query result buffer, and writes the result either * to a summary buffer to be consumed by the next grid invocation or to the * user-supplied buffer. * * Data layout: * * BUFFER[0] = query result buffer (layout is defined by gfx10_sh_query_buffer_mem) * BUFFER[1] = previous summary buffer * BUFFER[2] = next summary buffer or user-supplied buffer * * CONST * 0.x = config; the low 3 bits indicate the mode: * 0: sum up counts * 1: determine result availability and write it as a boolean * 2: SO_OVERFLOW * 3: SO_ANY_OVERFLOW * the remaining bits form a bitfield: * 8: write result as a 64-bit value * 0.y = offset in bytes to counts or stream for SO_OVERFLOW mode * 0.z = chain bit field: * 1: have previous summary buffer * 2: write next summary buffer * 0.w = result_count */ void *gfx10_create_sh_query_result_cs(struct si_context *sctx) { /* TEMP[0].x = accumulated result so far * TEMP[0].y = result missing * TEMP[0].z = whether we're in overflow mode */ static const char text_tmpl[] = "COMP\n" "PROPERTY CS_FIXED_BLOCK_WIDTH 1\n" "PROPERTY CS_FIXED_BLOCK_HEIGHT 1\n" "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" "DCL BUFFER[0]\n" "DCL BUFFER[1]\n" "DCL BUFFER[2]\n" "DCL CONST[0][0..0]\n" "DCL TEMP[0..5]\n" "IMM[0] UINT32 {0, 7, 0, 4294967295}\n" "IMM[1] UINT32 {1, 2, 4, 8}\n" "IMM[2] UINT32 {16, 32, 64, 128}\n" /* acc_result = 0; acc_missing = 0; if (chain & 1) { acc_result = buffer[1][0]; acc_missing = buffer[1][1]; } */ "MOV TEMP[0].xy, IMM[0].xxxx\n" "AND TEMP[5], CONST[0][0].zzzz, IMM[1].xxxx\n" "UIF TEMP[5]\n" "LOAD TEMP[0].xy, BUFFER[1], IMM[0].xxxx\n" "ENDIF\n" /* is_overflow (TEMP[0].z) = (config & 7) >= 2; result_remaining (TEMP[1].x) = (is_overflow && acc_result) ? 0 : result_count; base_offset (TEMP[1].y) = 0; for (;;) { if (!result_remaining) break; result_remaining--; */ "AND TEMP[5].x, CONST[0][0].xxxx, IMM[0].yyyy\n" "USGE TEMP[0].z, TEMP[5].xxxx, IMM[1].yyyy\n" "AND TEMP[5].x, TEMP[0].zzzz, TEMP[0].xxxx\n" "UCMP TEMP[1].x, TEMP[5].xxxx, IMM[0].xxxx, CONST[0][0].wwww\n" "MOV TEMP[1].y, IMM[0].xxxx\n" "BGNLOOP\n" "USEQ TEMP[5], TEMP[1].xxxx, IMM[0].xxxx\n" "UIF TEMP[5]\n" "BRK\n" "ENDIF\n" "UADD TEMP[1].x, TEMP[1].xxxx, IMM[0].wwww\n" /* fence = buffer[0]@(base_offset + 32); if (!fence) { acc_missing = ~0u; break; } */ "UADD TEMP[5].x, TEMP[1].yyyy, IMM[2].yyyy\n" "LOAD TEMP[5].x, BUFFER[0], TEMP[5].xxxx\n" "USEQ TEMP[5], TEMP[5].xxxx, IMM[0].xxxx\n" "UIF TEMP[5]\n" "MOV TEMP[0].y, TEMP[5].xxxx\n" "BRK\n" "ENDIF\n" /* stream_offset (TEMP[2].x) = base_offset + offset; if (!(config & 7)) { acc_result += buffer[0]@stream_offset; } */ "UADD TEMP[2].x, TEMP[1].yyyy, CONST[0][0].yyyy\n" "AND TEMP[5].x, CONST[0][0].xxxx, IMM[0].yyyy\n" "USEQ TEMP[5], TEMP[5].xxxx, IMM[0].xxxx\n" "UIF TEMP[5]\n" "LOAD TEMP[5].x, BUFFER[0], TEMP[2].xxxx\n" "UADD TEMP[0].x, TEMP[0].xxxx, TEMP[5].xxxx\n" "ENDIF\n" /* if ((config & 7) >= 2) { count (TEMP[2].y) = (config & 1) ? 4 : 1; */ "AND TEMP[5].x, CONST[0][0].xxxx, IMM[0].yyyy\n" "USGE TEMP[5], TEMP[5].xxxx, IMM[1].yyyy\n" "UIF TEMP[5]\n" "AND TEMP[5].x, CONST[0][0].xxxx, IMM[1].xxxx\n" "UCMP TEMP[2].y, TEMP[5].xxxx, IMM[1].zzzz, IMM[1].xxxx\n" /* do { generated = buffer[0]@stream_offset; emitted = buffer[0]@(stream_offset + 16); if (generated != emitted) { acc_result = 1; result_remaining = 0; break; } stream_offset += 4; } while (--count); */ "BGNLOOP\n" "UADD TEMP[5].x, TEMP[2].xxxx, IMM[2].xxxx\n" "LOAD TEMP[4].x, BUFFER[0], TEMP[2].xxxx\n" "LOAD TEMP[4].y, BUFFER[0], TEMP[5].xxxx\n" "USNE TEMP[5], TEMP[4].xxxx, TEMP[4].yyyy\n" "UIF TEMP[5]\n" "MOV TEMP[0].x, IMM[1].xxxx\n" "MOV TEMP[1].y, IMM[0].xxxx\n" "BRK\n" "ENDIF\n" "UADD TEMP[2].y, TEMP[2].yyyy, IMM[0].wwww\n" "USEQ TEMP[5], TEMP[2].yyyy, IMM[0].xxxx\n" "UIF TEMP[5]\n" "BRK\n" "ENDIF\n" "UADD TEMP[2].x, TEMP[2].xxxx, IMM[1].zzzz\n" "ENDLOOP\n" "ENDIF\n" /* base_offset += 64; } // end outer loop */ "UADD TEMP[1].y, TEMP[1].yyyy, IMM[2].zzzz\n" "ENDLOOP\n" /* if (chain & 2) { buffer[2][0] = acc_result; buffer[2][1] = acc_missing; } else { */ "AND TEMP[5], CONST[0][0].zzzz, IMM[1].yyyy\n" "UIF TEMP[5]\n" "STORE BUFFER[2].xy, IMM[0].xxxx, TEMP[0]\n" "ELSE\n" /* if ((config & 7) == 1) { acc_result = acc_missing ? 0 : 1; acc_missing = 0; } */ "AND TEMP[5], CONST[0][0].xxxx, IMM[0].yyyy\n" "USEQ TEMP[5], TEMP[5].xxxx, IMM[1].xxxx\n" "UIF TEMP[5]\n" "UCMP TEMP[0].x, TEMP[0].yyyy, IMM[0].xxxx, IMM[1].xxxx\n" "MOV TEMP[0].y, IMM[0].xxxx\n" "ENDIF\n" /* if (!acc_missing) { buffer[2][0] = acc_result; if (config & 8) buffer[2][1] = 0; } */ "USEQ TEMP[5], TEMP[0].yyyy, IMM[0].xxxx\n" "UIF TEMP[5]\n" "STORE BUFFER[2].x, IMM[0].xxxx, TEMP[0].xxxx\n" "AND TEMP[5], CONST[0][0].xxxx, IMM[1].wwww\n" "UIF TEMP[5]\n" "STORE BUFFER[2].x, IMM[1].zzzz, TEMP[0].yyyy\n" "ENDIF\n" "ENDIF\n" "ENDIF\n" "END\n"; struct tgsi_token tokens[1024]; struct pipe_compute_state state = {}; if (!tgsi_text_translate(text_tmpl, tokens, ARRAY_SIZE(tokens))) { assert(false); return NULL; } state.ir_type = PIPE_SHADER_IR_TGSI; state.prog = tokens; return sctx->b.create_compute_state(&sctx->b, &state); }