Home | History | Annotate | Line # | Download | only in compiler 
      1 /*
      2  * Copyright (C) 2021 Alyssa Rosenzweig <alyssa (at) rosenzweig.io>
      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 FROM,
     20  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
     21  * SOFTWARE.
     22  */
     23 
     24 #include "agx_compiler.h"
     25 
     26 /* Manages the uniform file. We can push certain fixed items (during initial
     27  * code generation), where we're gauranteed to have sufficient space. After
     28  * that, UBO ranges can be selectively pushed while there's space. */
     29 
     30 /* Directly index an array sysval. Index must be in bounds. Index specified in
     31  * 16-bit units regardless of the underlying sysval's unit. */
     32 
     33 agx_index
     34 agx_indexed_sysval(agx_context *ctx, enum agx_push_type type,
     35       enum agx_size size, unsigned index, unsigned length)
     36 {
     37    /* Check if we already pushed */
     38    for (unsigned i = 0; i < ctx->out->push_ranges; ++i) {
     39       struct agx_push push = ctx->out->push[i];
     40 
     41       if (push.type == type && !push.indirect) {
     42          assert(length == push.length);
     43          assert(index < push.length);
     44          return agx_uniform(push.base + index, size);
     45       }
     46    }
     47 
     48    /* Otherwise, push */
     49    assert(ctx->out->push_ranges < AGX_MAX_PUSH_RANGES);
     50 
     51    unsigned base = ctx->push_base;
     52    ctx->push_base += length;
     53 
     54    ctx->out->push[ctx->out->push_ranges++] = (struct agx_push) {
     55       .type = type,
     56       .base = base,
     57       .length = length,
     58       .indirect = false
     59    };
     60 
     61    return agx_uniform(base + index, size);
     62 }
     63