src/gallium/drivers/nouveau/nvc0/nve4_compute.c

   1 /*
   2  * Copyright 2012 Nouveau Project
   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 shall be included in
  12  * all copies or substantial portions of the Software.
  13  *
  14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  17  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
  18  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
  19  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
  20  * OTHER DEALINGS IN THE SOFTWARE.
  21  *
  22  * Authors: Christoph Bumiller
  23  */
  24
  25 #include "nvc0/nvc0_context.h"
  26 #include "nvc0/nve4_compute.h"
  27
  28 #include "codegen/nv50_ir_driver.h"
  29
  30 #ifdef DEBUG
  31 static void nve4_compute_dump_launch_desc(const struct nve4_cp_launch_desc *);
  32 #endif
  33
  34
  35 int
  36 nve4_screen_compute_setup(struct nvc0_screen *screen,
  37                           struct nouveau_pushbuf *push)
  38 {
  39    struct nouveau_device *dev = screen->base.device;
  40    struct nouveau_object *chan = screen->base.channel;
  41    int i;
  42    int ret;
  43    uint32_t obj_class;
  44    uint64_t address;
  45
  46    switch (dev->chipset & ~0xf) {
  47    case 0x100:
  48    case 0xf0:
  49       obj_class = NVF0_COMPUTE_CLASS; /* GK110 */
  50       break;
  51    case 0xe0:
  52       obj_class = NVE4_COMPUTE_CLASS; /* GK104 */
  53       break;
  54    case 0x110:
  55       obj_class = GM107_COMPUTE_CLASS;
  56       break;
  57    case 0x120:
  58       obj_class = GM200_COMPUTE_CLASS;
  59       break;
  60    default:
  61       NOUVEAU_ERR("unsupported chipset: NV%02x\n", dev->chipset);
  62       return -1;
  63    }
  64
  65    ret = nouveau_object_new(chan, 0xbeef00c0, obj_class, NULL, 0,
  66                             &screen->compute);
  67    if (ret) {
  68       NOUVEAU_ERR("Failed to allocate compute object: %d\n", ret);
  69       return ret;
  70    }
  71
  72    BEGIN_NVC0(push, SUBC_CP(NV01_SUBCHAN_OBJECT), 1);
  73    PUSH_DATA (push, screen->compute->oclass);
  74
  75    BEGIN_NVC0(push, NVE4_CP(TEMP_ADDRESS_HIGH), 2);
  76    PUSH_DATAh(push, screen->tls->offset);
  77    PUSH_DATA (push, screen->tls->offset);
  78    /* No idea why there are 2. Divide size by 2 to be safe.
  79     * Actually this might be per-MP TEMP size and looks like I'm only using
  80     * 2 MPs instead of all 8.
  81     */
  82    BEGIN_NVC0(push, NVE4_CP(MP_TEMP_SIZE_HIGH(0)), 3);
  83    PUSH_DATAh(push, screen->tls->size / screen->mp_count);
  84    PUSH_DATA (push, (screen->tls->size / screen->mp_count) & ~0x7fff);
  85    PUSH_DATA (push, 0xff);
  86    BEGIN_NVC0(push, NVE4_CP(MP_TEMP_SIZE_HIGH(1)), 3);
  87    PUSH_DATAh(push, screen->tls->size / screen->mp_count);
  88    PUSH_DATA (push, (screen->tls->size / screen->mp_count) & ~0x7fff);
  89    PUSH_DATA (push, 0xff);
  90
  91    /* Unified address space ? Who needs that ? Certainly not OpenCL.
  92     *
  93     * FATAL: Buffers with addresses inside [0x1000000, 0x3000000] will NOT be
  94     *  accessible. We cannot prevent that at the moment, so expect failure.
  95     */
  96    BEGIN_NVC0(push, NVE4_CP(LOCAL_BASE), 1);
  97    PUSH_DATA (push, 0xff << 24);
  98    BEGIN_NVC0(push, NVE4_CP(SHARED_BASE), 1);
  99    PUSH_DATA (push, 0xfe << 24);
 100
 101    BEGIN_NVC0(push, NVE4_CP(CODE_ADDRESS_HIGH), 2);
 102    PUSH_DATAh(push, screen->text->offset);
 103    PUSH_DATA (push, screen->text->offset);
 104
 105    BEGIN_NVC0(push, SUBC_CP(0x0310), 1);
 106    PUSH_DATA (push, (obj_class >= NVF0_COMPUTE_CLASS) ? 0x400 : 0x300);
 107
 108    /* NOTE: these do not affect the state used by the 3D object */
 109    BEGIN_NVC0(push, NVE4_CP(TIC_ADDRESS_HIGH), 3);
 110    PUSH_DATAh(push, screen->txc->offset);
 111    PUSH_DATA (push, screen->txc->offset);
 112    PUSH_DATA (push, NVC0_TIC_MAX_ENTRIES - 1);
 113    BEGIN_NVC0(push, NVE4_CP(TSC_ADDRESS_HIGH), 3);
 114    PUSH_DATAh(push, screen->txc->offset + 65536);
 115    PUSH_DATA (push, screen->txc->offset + 65536);
 116    PUSH_DATA (push, NVC0_TSC_MAX_ENTRIES - 1);
 117
 118    if (obj_class >= NVF0_COMPUTE_CLASS) {
 119       /* The blob calls GK110_COMPUTE.FIRMWARE[0x6], along with the args (0x1)
 120        * passed with GK110_COMPUTE.GRAPH.SCRATCH[0x2]. This is currently
 121        * disabled because our firmware doesn't support these commands and the
 122        * GPU hangs if they are used. */
 123       BEGIN_NIC0(push, SUBC_CP(0x0248), 64);
 124       for (i = 63; i >= 0; i--)
 125          PUSH_DATA(push, 0x38000 | i);
 126       IMMED_NVC0(push, SUBC_CP(NV50_GRAPH_SERIALIZE), 0);
 127    }
 128
 129    BEGIN_NVC0(push, NVE4_CP(TEX_CB_INDEX), 1);
 130    PUSH_DATA (push, 7); /* does not interfere with 3D */
 131
 132    /* Disabling this UNK command avoid a read fault when using texelFetch()
 133     * from a compute shader for weird reasons.
 134    if (obj_class == NVF0_COMPUTE_CLASS)
 135       IMMED_NVC0(push, SUBC_CP(0x02c4), 1);
 136    */
 137
 138    address = screen->uniform_bo->offset + NVC0_CB_AUX_INFO(5);
 139
 140    /* MS sample coordinate offsets: these do not work with _ALT modes ! */
 141    BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 142    PUSH_DATAh(push, address + NVC0_CB_AUX_MS_INFO);
 143    PUSH_DATA (push, address + NVC0_CB_AUX_MS_INFO);
 144    BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 145    PUSH_DATA (push, 64);
 146    PUSH_DATA (push, 1);
 147    BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 17);
 148    PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x20 << 1));
 149    PUSH_DATA (push, 0); /* 0 */
 150    PUSH_DATA (push, 0);
 151    PUSH_DATA (push, 1); /* 1 */
 152    PUSH_DATA (push, 0);
 153    PUSH_DATA (push, 0); /* 2 */
 154    PUSH_DATA (push, 1);
 155    PUSH_DATA (push, 1); /* 3 */
 156    PUSH_DATA (push, 1);
 157    PUSH_DATA (push, 2); /* 4 */
 158    PUSH_DATA (push, 0);
 159    PUSH_DATA (push, 3); /* 5 */
 160    PUSH_DATA (push, 0);
 161    PUSH_DATA (push, 2); /* 6 */
 162    PUSH_DATA (push, 1);
 163    PUSH_DATA (push, 3); /* 7 */
 164    PUSH_DATA (push, 1);
 165
 166 #ifdef NOUVEAU_NVE4_MP_TRAP_HANDLER
 167    BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 168    PUSH_DATAh(push, screen->parm->offset + NVE4_CP_INPUT_TRAP_INFO_PTR);
 169    PUSH_DATA (push, screen->parm->offset + NVE4_CP_INPUT_TRAP_INFO_PTR);
 170    BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 171    PUSH_DATA (push, 28);
 172    PUSH_DATA (push, 1);
 173    BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 8);
 174    PUSH_DATA (push, 1);
 175    PUSH_DATA (push, screen->parm->offset + NVE4_CP_PARAM_TRAP_INFO);
 176    PUSH_DATAh(push, screen->parm->offset + NVE4_CP_PARAM_TRAP_INFO);
 177    PUSH_DATA (push, screen->tls->offset);
 178    PUSH_DATAh(push, screen->tls->offset);
 179    PUSH_DATA (push, screen->tls->size / 2); /* MP TEMP block size */
 180    PUSH_DATA (push, screen->tls->size / 2 / 64); /* warp TEMP block size */
 181    PUSH_DATA (push, 0); /* warp cfstack size */
 182 #endif
 183
 184    BEGIN_NVC0(push, NVE4_CP(FLUSH), 1);
 185    PUSH_DATA (push, NVE4_COMPUTE_FLUSH_CB);
 186
 187    return 0;
 188 }
 189
 190 static void
 191 nve4_compute_validate_surfaces(struct nvc0_context *nvc0)
 192 {
 193    struct nouveau_pushbuf *push = nvc0->base.pushbuf;
 194    uint64_t address;
 195    const int s = 5;
 196    int i, j;
 197
 198    if (!nvc0->images_dirty[s])
 199       return;
 200
 201    address = nvc0->screen->uniform_bo->offset + NVC0_CB_AUX_INFO(s);
 202
 203    BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 204    PUSH_DATAh(push, address + NVC0_CB_AUX_SU_INFO(0));
 205    PUSH_DATA (push, address + NVC0_CB_AUX_SU_INFO(0));
 206    BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 207    PUSH_DATA (push, 16 * NVC0_MAX_IMAGES * 4);
 208    PUSH_DATA (push, 0x1);
 209    BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 1 + 16 * NVC0_MAX_IMAGES);
 210    PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x20 << 1));
 211
 212    for (i = 0; i < NVC0_MAX_IMAGES; ++i) {
 213       struct pipe_image_view *view = &nvc0->images[s][i];
 214       if (view->resource) {
 215          struct nv04_resource *res = nv04_resource(view->resource);
 216
 217          if (res->base.target == PIPE_BUFFER) {
 218             if (view->access & PIPE_IMAGE_ACCESS_WRITE)
 219                nvc0_mark_image_range_valid(view);
 220          }
 221
 222          nve4_set_surface_info(push, view, nvc0);
 223          BCTX_REFN(nvc0->bufctx_cp, CP_SUF, res, RDWR);
 224       } else {
 225          for (j = 0; j < 16; j++)
 226             PUSH_DATA(push, 0);
 227       }
 228    }
 229 }
 230
 231 /* Thankfully, textures with samplers follow the normal rules. */
 232 static void
 233 nve4_compute_validate_samplers(struct nvc0_context *nvc0)
 234 {
 235    bool need_flush = nve4_validate_tsc(nvc0, 5);
 236    if (need_flush) {
 237       BEGIN_NVC0(nvc0->base.pushbuf, NVE4_CP(TSC_FLUSH), 1);
 238       PUSH_DATA (nvc0->base.pushbuf, 0);
 239    }
 240
 241    /* Invalidate all 3D samplers because they are aliased. */
 242    for (int s = 0; s < 5; s++)
 243       nvc0->samplers_dirty[s] = ~0;
 244    nvc0->dirty_3d |= NVC0_NEW_3D_SAMPLERS;
 245 }
 246
 247 /* (Code duplicated at bottom for various non-convincing reasons.
 248  *  E.g. we might want to use the COMPUTE subchannel to upload TIC/TSC
 249  *  entries to avoid a subchannel switch.
 250  *  Same for texture cache flushes.
 251  *  Also, the bufctx differs, and more IFs in the 3D version looks ugly.)
 252  */
 253 static void nve4_compute_validate_textures(struct nvc0_context *);
 254
 255 static void
 256 nve4_compute_set_tex_handles(struct nvc0_context *nvc0)
 257 {
 258    struct nouveau_pushbuf *push = nvc0->base.pushbuf;
 259    struct nvc0_screen *screen = nvc0->screen;
 260    uint64_t address;
 261    const unsigned s = nvc0_shader_stage(PIPE_SHADER_COMPUTE);
 262    unsigned i, n;
 263    uint32_t dirty = nvc0->textures_dirty[s] | nvc0->samplers_dirty[s];
 264
 265    if (!dirty)
 266       return;
 267    i = ffs(dirty) - 1;
 268    n = util_logbase2(dirty) + 1 - i;
 269    assert(n);
 270
 271    address = screen->uniform_bo->offset + NVC0_CB_AUX_INFO(s);
 272
 273    BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 274    PUSH_DATAh(push, address + NVC0_CB_AUX_TEX_INFO(i));
 275    PUSH_DATA (push, address + NVC0_CB_AUX_TEX_INFO(i));
 276    BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 277    PUSH_DATA (push, n * 4);
 278    PUSH_DATA (push, 0x1);
 279    BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 1 + n);
 280    PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x20 << 1));
 281    PUSH_DATAp(push, &nvc0->tex_handles[s][i], n);
 282
 283    BEGIN_NVC0(push, NVE4_CP(FLUSH), 1);
 284    PUSH_DATA (push, NVE4_COMPUTE_FLUSH_CB);
 285
 286    nvc0->textures_dirty[s] = 0;
 287    nvc0->samplers_dirty[s] = 0;
 288 }
 289
 290 static void
 291 nve4_compute_validate_constbufs(struct nvc0_context *nvc0)
 292 {
 293    struct nouveau_pushbuf *push = nvc0->base.pushbuf;
 294    const int s = 5;
 295
 296    while (nvc0->constbuf_dirty[s]) {
 297       int i = ffs(nvc0->constbuf_dirty[s]) - 1;
 298       nvc0->constbuf_dirty[s] &= ~(1 << i);
 299
 300       if (nvc0->constbuf[s][i].user) {
 301          struct nouveau_bo *bo = nvc0->screen->uniform_bo;
 302          const unsigned base = NVC0_CB_USR_INFO(s);
 303          const unsigned size = nvc0->constbuf[s][0].size;
 304          assert(i == 0); /* we really only want OpenGL uniforms here */
 305          assert(nvc0->constbuf[s][0].u.data);
 306
 307          BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 308          PUSH_DATAh(push, bo->offset + base);
 309          PUSH_DATA (push, bo->offset + base);
 310          BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 311          PUSH_DATA (push, size);
 312          PUSH_DATA (push, 0x1);
 313          BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 1 + (size / 4));
 314          PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x20 << 1));
 315          PUSH_DATAp(push, nvc0->constbuf[s][0].u.data, size / 4);
 316       }
 317       else {
 318          struct nv04_resource *res =
 319             nv04_resource(nvc0->constbuf[s][i].u.buf);
 320          if (res) {
 321             uint64_t address
 322                = nvc0->screen->uniform_bo->offset + NVC0_CB_AUX_INFO(s);
 323
 324             assert(i > 0); /* we really only want uniform buffer objects */
 325
 326             BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 327             PUSH_DATAh(push, address + NVC0_CB_AUX_UBO_INFO(i - 1));
 328             PUSH_DATA (push, address + NVC0_CB_AUX_UBO_INFO(i - 1));
 329             BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 330             PUSH_DATA (push, 4 * 4);
 331             PUSH_DATA (push, 0x1);
 332             BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 1 + 4);
 333             PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x20 << 1));
 334
 335             PUSH_DATA (push, res->address + nvc0->constbuf[s][i].offset);
 336             PUSH_DATAh(push, res->address + nvc0->constbuf[s][i].offset);
 337             PUSH_DATA (push, nvc0->constbuf[5][i].size);
 338             PUSH_DATA (push, 0);
 339             BCTX_REFN(nvc0->bufctx_cp, CP_CB(i), res, RD);
 340
 341             res->cb_bindings[s] |= 1 << i;
 342          }
 343       }
 344    }
 345
 346    BEGIN_NVC0(push, NVE4_CP(FLUSH), 1);
 347    PUSH_DATA (push, NVE4_COMPUTE_FLUSH_CB);
 348 }
 349
 350 static void
 351 nve4_compute_validate_buffers(struct nvc0_context *nvc0)
 352 {
 353    struct nouveau_pushbuf *push = nvc0->base.pushbuf;
 354    uint64_t address;
 355    const int s = 5;
 356    int i;
 357
 358    address = nvc0->screen->uniform_bo->offset + NVC0_CB_AUX_INFO(s);
 359
 360    BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 361    PUSH_DATAh(push, address + NVC0_CB_AUX_BUF_INFO(0));
 362    PUSH_DATA (push, address + NVC0_CB_AUX_BUF_INFO(0));
 363    BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 364    PUSH_DATA (push, 4 * NVC0_MAX_BUFFERS * 4);
 365    PUSH_DATA (push, 0x1);
 366    BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 1 + 4 * NVC0_MAX_BUFFERS);
 367    PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x20 << 1));
 368
 369    for (i = 0; i < NVC0_MAX_BUFFERS; i++) {
 370       if (nvc0->buffers[s][i].buffer) {
 371          struct nv04_resource *res =
 372             nv04_resource(nvc0->buffers[s][i].buffer);
 373          PUSH_DATA (push, res->address + nvc0->buffers[s][i].buffer_offset);
 374          PUSH_DATAh(push, res->address + nvc0->buffers[s][i].buffer_offset);
 375          PUSH_DATA (push, nvc0->buffers[s][i].buffer_size);
 376          PUSH_DATA (push, 0);
 377          BCTX_REFN(nvc0->bufctx_cp, CP_BUF, res, RDWR);
 378          util_range_add(&res->valid_buffer_range,
 379                         nvc0->buffers[s][i].buffer_offset,
 380                         nvc0->buffers[s][i].buffer_size);
 381       } else {
 382          PUSH_DATA (push, 0);
 383          PUSH_DATA (push, 0);
 384          PUSH_DATA (push, 0);
 385          PUSH_DATA (push, 0);
 386       }
 387    }
 388 }
 389
 390 static struct nvc0_state_validate
 391 validate_list_cp[] = {
 392    { nvc0_compprog_validate,              NVC0_NEW_CP_PROGRAM     },
 393    { nve4_compute_validate_textures,      NVC0_NEW_CP_TEXTURES    },
 394    { nve4_compute_validate_samplers,      NVC0_NEW_CP_SAMPLERS    },
 395    { nve4_compute_set_tex_handles,        NVC0_NEW_CP_TEXTURES |
 396                                           NVC0_NEW_CP_SAMPLERS    },
 397    { nve4_compute_validate_surfaces,      NVC0_NEW_CP_SURFACES    },
 398    { nvc0_compute_validate_globals,       NVC0_NEW_CP_GLOBALS     },
 399    { nve4_compute_validate_buffers,       NVC0_NEW_CP_BUFFERS     },
 400    { nve4_compute_validate_constbufs,     NVC0_NEW_CP_CONSTBUF    },
 401 };
 402
 403 static bool
 404 nve4_state_validate_cp(struct nvc0_context *nvc0, uint32_t mask)
 405 {
 406    bool ret;
 407
 408    ret = nvc0_state_validate(nvc0, mask, validate_list_cp,
 409                              ARRAY_SIZE(validate_list_cp), &nvc0->dirty_cp,
 410                              nvc0->bufctx_cp);
 411
 412    if (unlikely(nvc0->state.flushed))
 413       nvc0_bufctx_fence(nvc0, nvc0->bufctx_cp, true);
 414    return ret;
 415 }
 416
 417 static void
 418 nve4_compute_upload_input(struct nvc0_context *nvc0,
 419                           const struct pipe_grid_info *info)
 420 {
 421    struct nvc0_screen *screen = nvc0->screen;
 422    struct nouveau_pushbuf *push = nvc0->base.pushbuf;
 423    struct nvc0_program *cp = nvc0->compprog;
 424    uint64_t address;
 425
 426    address = screen->uniform_bo->offset + NVC0_CB_AUX_INFO(5);
 427
 428    if (cp->parm_size) {
 429       BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 430       PUSH_DATAh(push, screen->uniform_bo->offset + NVC0_CB_USR_INFO(5));
 431       PUSH_DATA (push, screen->uniform_bo->offset + NVC0_CB_USR_INFO(5));
 432       BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 433       PUSH_DATA (push, cp->parm_size);
 434       PUSH_DATA (push, 0x1);
 435       BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 1 + (cp->parm_size / 4));
 436       PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x20 << 1));
 437       PUSH_DATAp(push, info->input, cp->parm_size / 4);
 438    }
 439    BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 440    PUSH_DATAh(push, address + NVC0_CB_AUX_GRID_INFO);
 441    PUSH_DATA (push, address + NVC0_CB_AUX_GRID_INFO);
 442    BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 443    PUSH_DATA (push, 7 * 4);
 444    PUSH_DATA (push, 0x1);
 445
 446    if (unlikely(info->indirect)) {
 447       struct nv04_resource *res = nv04_resource(info->indirect);
 448       uint32_t offset = res->offset + info->indirect_offset;
 449
 450       nouveau_pushbuf_space(push, 16, 0, 1);
 451       PUSH_REFN(push, res->bo, NOUVEAU_BO_RD | res->domain);
 452
 453       BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 1 + 7);
 454       PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x20 << 1));
 455       PUSH_DATAp(push, info->block, 3);
 456       nouveau_pushbuf_data(push, res->bo, offset,
 457                            NVC0_IB_ENTRY_1_NO_PREFETCH | 3 * 4);
 458    } else {
 459       BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 1 + 7);
 460       PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x20 << 1));
 461       PUSH_DATAp(push, info->block, 3);
 462       PUSH_DATAp(push, info->grid, 3);
 463    }
 464    PUSH_DATA (push, 0);
 465
 466    BEGIN_NVC0(push, NVE4_CP(FLUSH), 1);
 467    PUSH_DATA (push, NVE4_COMPUTE_FLUSH_CB);
 468 }
 469
 470 static inline uint8_t
 471 nve4_compute_derive_cache_split(struct nvc0_context *nvc0, uint32_t shared_size)
 472 {
 473    if (shared_size > (32 << 10))
 474       return NVC0_3D_CACHE_SPLIT_48K_SHARED_16K_L1;
 475    if (shared_size > (16 << 10))
 476       return NVE4_3D_CACHE_SPLIT_32K_SHARED_32K_L1;
 477    return NVC1_3D_CACHE_SPLIT_16K_SHARED_48K_L1;
 478 }
 479
 480 static void
 481 nve4_compute_setup_launch_desc(struct nvc0_context *nvc0,
 482                                struct nve4_cp_launch_desc *desc,
 483                                const struct pipe_grid_info *info)
 484 {
 485    const struct nvc0_screen *screen = nvc0->screen;
 486    const struct nvc0_program *cp = nvc0->compprog;
 487
 488    nve4_cp_launch_desc_init_default(desc);
 489
 490    desc->entry = nvc0_program_symbol_offset(cp, info->pc);
 491
 492    desc->griddim_x = info->grid[0];
 493    desc->griddim_y = info->grid[1];
 494    desc->griddim_z = info->grid[2];
 495    desc->blockdim_x = info->block[0];
 496    desc->blockdim_y = info->block[1];
 497    desc->blockdim_z = info->block[2];
 498
 499    desc->shared_size = align(cp->cp.smem_size, 0x100);
 500    desc->local_size_p = (cp->hdr[1] & 0xfffff0) + align(cp->cp.lmem_size, 0x10);
 501    desc->local_size_n = 0;
 502    desc->cstack_size = 0x800;
 503    desc->cache_split = nve4_compute_derive_cache_split(nvc0, cp->cp.smem_size);
 504
 505    desc->gpr_alloc = cp->num_gprs;
 506    desc->bar_alloc = cp->num_barriers;
 507
 508    // Only bind user uniforms and the driver constant buffer through the
 509    // launch descriptor because UBOs are sticked to the driver cb to avoid the
 510    // limitation of 8 CBs.
 511    if (nvc0->constbuf[5][0].user || cp->parm_size) {
 512       nve4_cp_launch_desc_set_cb(desc, 0, screen->uniform_bo,
 513                                  NVC0_CB_USR_INFO(5), 1 << 16);
 514    }
 515    nve4_cp_launch_desc_set_cb(desc, 7, screen->uniform_bo,
 516                               NVC0_CB_AUX_INFO(5), 1 << 11);
 517 }
 518
 519 static inline struct nve4_cp_launch_desc *
 520 nve4_compute_alloc_launch_desc(struct nouveau_context *nv,
 521                                struct nouveau_bo **pbo, uint64_t *pgpuaddr)
 522 {
 523    uint8_t *ptr = nouveau_scratch_get(nv, 512, pgpuaddr, pbo);
 524    if (!ptr)
 525       return NULL;
 526    if (*pgpuaddr & 255) {
 527       unsigned adj = 256 - (*pgpuaddr & 255);
 528       ptr += adj;
 529       *pgpuaddr += adj;
 530    }
 531    return (struct nve4_cp_launch_desc *)ptr;
 532 }
 533
 534 void
 535 nve4_launch_grid(struct pipe_context *pipe, const struct pipe_grid_info *info)
 536 {
 537    struct nvc0_context *nvc0 = nvc0_context(pipe);
 538    struct nvc0_screen *screen = nvc0->screen;
 539    struct nouveau_pushbuf *push = nvc0->base.pushbuf;
 540    struct nve4_cp_launch_desc *desc;
 541    uint64_t desc_gpuaddr;
 542    struct nouveau_bo *desc_bo;
 543    int ret;
 544
 545    pipe_mutex_lock(screen->base.push_mutex);
 546
 547    desc = nve4_compute_alloc_launch_desc(&nvc0->base, &desc_bo, &desc_gpuaddr);
 548    if (!desc) {
 549       ret = -1;
 550       goto out;
 551    }
 552    BCTX_REFN_bo(nvc0->bufctx_cp, CP_DESC, NOUVEAU_BO_GART | NOUVEAU_BO_RD,
 553                 desc_bo);
 554
 555    ret = !nve4_state_validate_cp(nvc0, ~0);
 556    if (ret)
 557       goto out;
 558
 559    nve4_compute_setup_launch_desc(nvc0, desc, info);
 560
 561    nve4_compute_upload_input(nvc0, info);
 562
 563 #ifdef DEBUG
 564    if (debug_get_num_option("NV50_PROG_DEBUG", 0))
 565       nve4_compute_dump_launch_desc(desc);
 566 #endif
 567
 568    if (unlikely(info->indirect)) {
 569       struct nv04_resource *res = nv04_resource(info->indirect);
 570       uint32_t offset = res->offset + info->indirect_offset;
 571
 572       /* upload the descriptor */
 573       BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 574       PUSH_DATAh(push, desc_gpuaddr);
 575       PUSH_DATA (push, desc_gpuaddr);
 576       BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 577       PUSH_DATA (push, 256);
 578       PUSH_DATA (push, 1);
 579       BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 1 + (256 / 4));
 580       PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x08 << 1));
 581       PUSH_DATAp(push, (const uint32_t *)desc, 256 / 4);
 582
 583       /* overwrite griddim_x and griddim_y as two 32-bits integers even
 584        * if griddim_y must be a 16-bits integer */
 585       BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 586       PUSH_DATAh(push, desc_gpuaddr + 48);
 587       PUSH_DATA (push, desc_gpuaddr + 48);
 588       BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 589       PUSH_DATA (push, 8);
 590       PUSH_DATA (push, 1);
 591
 592       nouveau_pushbuf_space(push, 16, 0, 1);
 593       PUSH_REFN(push, res->bo, NOUVEAU_BO_RD | res->domain);
 594
 595       BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 1 + (8 / 4));
 596       PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x08 << 1));
 597       nouveau_pushbuf_data(push, res->bo, offset,
 598                            NVC0_IB_ENTRY_1_NO_PREFETCH | 2 * 4);
 599
 600       /* overwrite the 16 high bits of griddim_y with griddim_z because
 601        * we need (z << 16) | x */
 602       BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 603       PUSH_DATAh(push, desc_gpuaddr + 54);
 604       PUSH_DATA (push, desc_gpuaddr + 54);
 605       BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 606       PUSH_DATA (push, 4);
 607       PUSH_DATA (push, 1);
 608       BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 1 + (4 / 4));
 609       PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x08 << 1));
 610       nouveau_pushbuf_data(push, res->bo, offset + 8,
 611                            NVC0_IB_ENTRY_1_NO_PREFETCH | 1 * 4);
 612    }
 613
 614    /* upload descriptor and flush */
 615    BEGIN_NVC0(push, NVE4_CP(LAUNCH_DESC_ADDRESS), 1);
 616    PUSH_DATA (push, desc_gpuaddr >> 8);
 617    BEGIN_NVC0(push, NVE4_CP(LAUNCH), 1);
 618    PUSH_DATA (push, 0x3);
 619    BEGIN_NVC0(push, SUBC_CP(NV50_GRAPH_SERIALIZE), 1);
 620    PUSH_DATA (push, 0);
 621
 622 out:
 623    if (ret)
 624       NOUVEAU_ERR("Failed to launch grid !\n");
 625    nouveau_scratch_done(&nvc0->base);
 626    nouveau_bufctx_reset(nvc0->bufctx_cp, NVC0_BIND_CP_DESC);
 627    pipe_mutex_unlock(screen->base.push_mutex);
 628 }
 629
 630
 631 #define NVE4_TIC_ENTRY_INVALID 0x000fffff
 632
 633 static void
 634 nve4_compute_validate_textures(struct nvc0_context *nvc0)
 635 {
 636    struct nouveau_bo *txc = nvc0->screen->txc;
 637    struct nouveau_pushbuf *push = nvc0->base.pushbuf;
 638    const unsigned s = 5;
 639    unsigned i;
 640    uint32_t commands[2][32];
 641    unsigned n[2] = { 0, 0 };
 642
 643    for (i = 0; i < nvc0->num_textures[s]; ++i) {
 644       struct nv50_tic_entry *tic = nv50_tic_entry(nvc0->textures[s][i]);
 645       struct nv04_resource *res;
 646       const bool dirty = !!(nvc0->textures_dirty[s] & (1 << i));
 647
 648       if (!tic) {
 649          nvc0->tex_handles[s][i] |= NVE4_TIC_ENTRY_INVALID;
 650          continue;
 651       }
 652       res = nv04_resource(tic->pipe.texture);
 653       nvc0_update_tic(nvc0, tic, res);
 654
 655       if (tic->id < 0) {
 656          tic->id = nvc0_screen_tic_alloc(nvc0->screen, tic);
 657
 658          PUSH_SPACE(push, 16);
 659          BEGIN_NVC0(push, NVE4_CP(UPLOAD_DST_ADDRESS_HIGH), 2);
 660          PUSH_DATAh(push, txc->offset + (tic->id * 32));
 661          PUSH_DATA (push, txc->offset + (tic->id * 32));
 662          BEGIN_NVC0(push, NVE4_CP(UPLOAD_LINE_LENGTH_IN), 2);
 663          PUSH_DATA (push, 32);
 664          PUSH_DATA (push, 1);
 665          BEGIN_1IC0(push, NVE4_CP(UPLOAD_EXEC), 9);
 666          PUSH_DATA (push, NVE4_COMPUTE_UPLOAD_EXEC_LINEAR | (0x20 << 1));
 667          PUSH_DATAp(push, &tic->tic[0], 8);
 668
 669          commands[0][n[0]++] = (tic->id << 4) | 1;
 670       } else
 671       if (res->status & NOUVEAU_BUFFER_STATUS_GPU_WRITING) {
 672          commands[1][n[1]++] = (tic->id << 4) | 1;
 673       }
 674       nvc0->screen->tic.lock[tic->id / 32] |= 1 << (tic->id % 32);
 675
 676       res->status &= ~NOUVEAU_BUFFER_STATUS_GPU_WRITING;
 677       res->status |=  NOUVEAU_BUFFER_STATUS_GPU_READING;
 678
 679       nvc0->tex_handles[s][i] &= ~NVE4_TIC_ENTRY_INVALID;
 680       nvc0->tex_handles[s][i] |= tic->id;
 681       if (dirty)
 682          BCTX_REFN(nvc0->bufctx_cp, CP_TEX(i), res, RD);
 683    }
 684    for (; i < nvc0->state.num_textures[s]; ++i) {
 685       nvc0->tex_handles[s][i] |= NVE4_TIC_ENTRY_INVALID;
 686       nvc0->textures_dirty[s] |= 1 << i;
 687    }
 688
 689    if (n[0]) {
 690       BEGIN_NIC0(push, NVE4_CP(TIC_FLUSH), n[0]);
 691       PUSH_DATAp(push, commands[0], n[0]);
 692    }
 693    if (n[1]) {
 694       BEGIN_NIC0(push, NVE4_CP(TEX_CACHE_CTL), n[1]);
 695       PUSH_DATAp(push, commands[1], n[1]);
 696    }
 697
 698    nvc0->state.num_textures[s] = nvc0->num_textures[s];
 699
 700    /* Invalidate all 3D textures because they are aliased. */
 701    for (int s = 0; s < 5; s++) {
 702       for (int i = 0; i < nvc0->num_textures[s]; i++)
 703          nouveau_bufctx_reset(nvc0->bufctx_3d, NVC0_BIND_3D_TEX(s, i));
 704       nvc0->textures_dirty[s] = ~0;
 705    }
 706    nvc0->dirty_3d |= NVC0_NEW_3D_TEXTURES;
 707 }
 708
 709
 710 #ifdef DEBUG
 711 static const char *nve4_cache_split_name(unsigned value)
 712 {
 713    switch (value) {
 714    case NVC1_3D_CACHE_SPLIT_16K_SHARED_48K_L1: return "16K_SHARED_48K_L1";
 715    case NVE4_3D_CACHE_SPLIT_32K_SHARED_32K_L1: return "32K_SHARED_32K_L1";
 716    case NVC0_3D_CACHE_SPLIT_48K_SHARED_16K_L1: return "48K_SHARED_16K_L1";
 717    default:
 718       return "(invalid)";
 719    }
 720 }
 721
 722 static void
 723 nve4_compute_dump_launch_desc(const struct nve4_cp_launch_desc *desc)
 724 {
 725    const uint32_t *data = (const uint32_t *)desc;
 726    unsigned i;
 727    bool zero = false;
 728
 729    debug_printf("COMPUTE LAUNCH DESCRIPTOR:\n");
 730
 731    for (i = 0; i < sizeof(*desc); i += 4) {
 732       if (data[i / 4]) {
 733          debug_printf("[%x]: 0x%08x\n", i, data[i / 4]);
 734          zero = false;
 735       } else
 736       if (!zero) {
 737          debug_printf("...\n");
 738          zero = true;
 739       }
 740    }
 741
 742    debug_printf("entry = 0x%x\n", desc->entry);
 743    debug_printf("grid dimensions = %ux%ux%u\n",
 744                 desc->griddim_x, desc->griddim_y, desc->griddim_z);
 745    debug_printf("block dimensions = %ux%ux%u\n",
 746                 desc->blockdim_x, desc->blockdim_y, desc->blockdim_z);
 747    debug_printf("s[] size: 0x%x\n", desc->shared_size);
 748    debug_printf("l[] size: -0x%x / +0x%x\n",
 749                 desc->local_size_n, desc->local_size_p);
 750    debug_printf("stack size: 0x%x\n", desc->cstack_size);
 751    debug_printf("barrier count: %u\n", desc->bar_alloc);
 752    debug_printf("$r count: %u\n", desc->gpr_alloc);
 753    debug_printf("cache split: %s\n", nve4_cache_split_name(desc->cache_split));
 754
 755    for (i = 0; i < 8; ++i) {
 756       uint64_t address;
 757       uint32_t size = desc->cb[i].size;
 758       bool valid = !!(desc->cb_mask & (1 << i));
 759
 760       address = ((uint64_t)desc->cb[i].address_h << 32) | desc->cb[i].address_l;
 761
 762       if (!valid && !address && !size)
 763          continue;
 764       debug_printf("CB[%u]: address = 0x%"PRIx64", size 0x%x%s\n",
 765                    i, address, size, valid ? "" : "  (invalid)");
 766    }
 767 }
 768 #endif
 769
 770 #ifdef NOUVEAU_NVE4_MP_TRAP_HANDLER
 771 static void
 772 nve4_compute_trap_info(struct nvc0_context *nvc0)
 773 {
 774    struct nvc0_screen *screen = nvc0->screen;
 775    struct nouveau_bo *bo = screen->parm;
 776    int ret, i;
 777    volatile struct nve4_mp_trap_info *info;
 778    uint8_t *map;
 779
 780    ret = nouveau_bo_map(bo, NOUVEAU_BO_RDWR, nvc0->base.client);
 781    if (ret)
 782       return;
 783    map = (uint8_t *)bo->map;
 784    info = (volatile struct nve4_mp_trap_info *)(map + NVE4_CP_PARAM_TRAP_INFO);
 785
 786    if (info->lock) {
 787       debug_printf("trapstat = %08x\n", info->trapstat);
 788       debug_printf("warperr = %08x\n", info->warperr);
 789       debug_printf("PC = %x\n", info->pc);
 790       debug_printf("tid = %u %u %u\n",
 791                    info->tid[0], info->tid[1], info->tid[2]);
 792       debug_printf("ctaid = %u %u %u\n",
 793                    info->ctaid[0], info->ctaid[1], info->ctaid[2]);
 794       for (i = 0; i <= 63; ++i)
 795          debug_printf("$r%i = %08x\n", i, info->r[i]);
 796       for (i = 0; i <= 6; ++i)
 797          debug_printf("$p%i = %i\n", i, (info->flags >> i) & 1);
 798       debug_printf("$c = %x\n", info->flags >> 12);
 799    }
 800    info->lock = 0;
 801 }
 802 #endif