src/gallium/drivers/llvmpipe/lp_state_cs.h

   1 /**************************************************************************
   2  *
   3  * Copyright 2019 Red Hat.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the "Software"),
   8  * to deal in the Software without restriction, including without limitation
   9  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  10  * and/or sell copies of the Software, and to permit persons to whom the
  11  * Software is furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included
  14  * in all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  17  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  19  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  22  * SOFTWARE.
  23  *
  24  **************************************************************************/
  25
  26 #ifndef LP_STATE_CS_H
  27 #define LP_STATE_CS_H
  28
  29 #include "os/os_thread.h"
  30 #include "util/u_thread.h"
  31 #include "pipe/p_state.h"
  32
  33 #include "gallivm/lp_bld.h"
  34 #include "gallivm/lp_bld_sample.h" /* for struct lp_sampler_static_state */
  35 #include "lp_jit.h"
  36 #include "lp_state_fs.h"
  37
  38 struct lp_compute_shader_variant;
  39
  40 struct lp_compute_shader_variant_key
  41 {
  42    unsigned nr_samplers:8;
  43    unsigned nr_sampler_views:8;
  44    unsigned nr_images:8;
  45    struct lp_image_static_state image_state[PIPE_MAX_SHADER_IMAGES];
  46    struct lp_sampler_static_state state[PIPE_MAX_SHADER_SAMPLER_VIEWS];
  47 };
  48
  49 struct lp_cs_variant_list_item
  50 {
  51    struct lp_compute_shader_variant *base;
  52    struct lp_cs_variant_list_item *next, *prev;
  53 };
  54
  55 struct lp_compute_shader_variant
  56 {
  57    struct lp_compute_shader_variant_key key;
  58
  59    struct gallivm_state *gallivm;
  60
  61    LLVMTypeRef jit_cs_context_ptr_type;
  62    LLVMTypeRef jit_cs_thread_data_ptr_type;
  63
  64    LLVMValueRef function;
  65    lp_jit_cs_func jit_function;
  66
  67    /* Total number of LLVM instructions generated */
  68    unsigned nr_instrs;
  69
  70    struct lp_cs_variant_list_item list_item_global, list_item_local;
  71
  72    struct lp_compute_shader *shader;
  73
  74    /* For debugging/profiling purposes */
  75    unsigned no;
  76 };
  77
  78 struct lp_compute_shader {
  79    struct pipe_shader_state base;
  80
  81    struct lp_cs_variant_list_item variants;
  82
  83    struct lp_tgsi_info info;
  84
  85    uint32_t req_local_mem;
  86
  87    /* For debugging/profiling purposes */
  88    unsigned variant_key_size;
  89    unsigned no;
  90    unsigned variants_created;
  91    unsigned variants_cached;
  92 };
  93
  94 struct lp_cs_exec {
  95    struct lp_jit_cs_context jit_context;
  96    struct lp_compute_shader_variant *variant;
  97 };
  98
  99 struct lp_cs_context {
 100    struct pipe_context *pipe;
 101
 102    struct {
 103       struct lp_cs_exec current;
 104       struct pipe_resource *current_tex[PIPE_MAX_SHADER_SAMPLER_VIEWS];
 105       unsigned current_tex_num;
 106    } cs;
 107
 108    /** compute shader constants */
 109    struct {
 110       struct pipe_constant_buffer current;
 111       unsigned stored_size;
 112       const void *stored_data;
 113    } constants[LP_MAX_TGSI_CONST_BUFFERS];
 114
 115    /** compute shader buffers */
 116    struct {
 117       struct pipe_shader_buffer current;
 118    } ssbos[LP_MAX_TGSI_SHADER_BUFFERS];
 119
 120    struct {
 121       struct pipe_image_view current;
 122    } images[LP_MAX_TGSI_SHADER_IMAGES];
 123 };
 124
 125 struct lp_cs_context *lp_csctx_create(struct pipe_context *pipe);
 126 void lp_csctx_destroy(struct lp_cs_context *csctx);
 127
 128 #endif