src/compiler/nir/nir_schedule.h

   1 /*
   2  * Copyright © 2014 Connor Abbott
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  21  * IN THE SOFTWARE.
  22  */
  23
  24 #ifndef NIR_SCHEDULE_H
  25 #define NIR_SCHEDULE_H
  26
  27 #include "nir.h"
  28
  29 #ifdef __cplusplus
  30 extern "C" {
  31 #endif
  32
  33 /**
  34  * Struct filled in by the intrinsic_cb callback of nir_schedule_options to
  35  * specify a backend-specific dependency on an intrinsic.
  36  */
  37 typedef struct nir_schedule_dependency {
  38    /* Which class of dependency this is. The meanings of the classes are
  39     * specific to the backend. This must be less than
  40     * NIR_SCHEDULE_N_DEPENDENCY_CLASSES.
  41     */
  42    int klass;
  43    /* The type of dependency */
  44    enum {
  45       NIR_SCHEDULE_READ_DEPENDENCY,
  46       NIR_SCHEDULE_WRITE_DEPENDENCY,
  47    } type;
  48 } nir_schedule_dependency;
  49
  50 typedef struct nir_schedule_options {
  51    /* On some hardware with some stages the inputs and outputs to the shader
  52     * share the same memory. In that case the scheduler needs to ensure that
  53     * all output writes are scheduled after all of the input writes to avoid
  54     * overwriting them. This is a bitmask of stages that need that.
  55     */
  56    unsigned stages_with_shared_io_memory;
  57    /* The approximate amount of register pressure at which point the scheduler
  58     * will try to reduce register usage.
  59     */
  60    int threshold;
  61    /* Callback used to add custom dependencies on intrinsics. If it returns
  62     * true then a dependency should be added and dep is filled in to describe
  63     * it.
  64     */
  65    bool (* intrinsic_cb)(nir_intrinsic_instr *intr,
  66                          nir_schedule_dependency *dep,
  67                          void *user_data);
  68    /* Data to pass to the callback */
  69    void *intrinsic_cb_data;
  70 } nir_schedule_options;
  71
  72 void nir_schedule(nir_shader *shader, const nir_schedule_options *options);
  73
  74 #ifdef __cplusplus
  75 } /* extern "C" */
  76 #endif
  77
  78 #endif /* NIR_SCHEDULE_H */