xref: /aosp_15_r20/external/mesa3d/src/intel/compiler/brw_ir_allocator.h (revision 6104692788411f58d303aa86923a9ff6ecaded22)
1 /* -*- c++ -*- */
2 /*
3  * Copyright © 2010-2014 Intel Corporation
4  *
5  * Permission is hereby granted, free of charge, to any person obtaining a
6  * copy of this software and associated documentation files (the "Software"),
7  * to deal in the Software without restriction, including without limitation
8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9  * and/or sell copies of the Software, and to permit persons to whom the
10  * Software is furnished to do so, subject to the following conditions:
11  *
12  * The above copyright notice and this permission notice (including the next
13  * paragraph) shall be included in all copies or substantial portions of the
14  * Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * 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
21  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
22  * IN THE SOFTWARE.
23  */
24 
25 #ifndef BRW_IR_ALLOCATOR_H
26 #define BRW_IR_ALLOCATOR_H
27 
28 #include "util/compiler.h"
29 #include "util/glheader.h"
30 #include "util/macros.h"
31 #include "util/rounding.h"
32 #include "util/u_math.h"
33 
34 namespace brw {
35    /**
36     * Simple allocator used to keep track of virtual GRFs.
37     */
38    class simple_allocator {
39    public:
simple_allocator()40       simple_allocator() :
41          sizes(NULL), offsets(NULL), count(0), total_size(0), capacity(0)
42       {
43       }
44 
~simple_allocator()45       ~simple_allocator()
46       {
47          free(offsets);
48          free(sizes);
49       }
50 
51       unsigned
allocate(unsigned size)52       allocate(unsigned size)
53       {
54          assert(size > 0);
55          if (capacity <= count) {
56             capacity = MAX2(16, capacity * 2);
57             sizes = (unsigned *)realloc(sizes, capacity * sizeof(unsigned));
58             offsets = (unsigned *)realloc(offsets, capacity * sizeof(unsigned));
59          }
60 
61          sizes[count] = size;
62          offsets[count] = total_size;
63          total_size += size;
64 
65          return count++;
66       }
67 
68       /**
69        * Array of sizes for each allocation.  The allocation unit is up to the
70        * back-end, but it's expected to be one scalar value in the FS back-end
71        * and one vec4 in the VEC4 back-end.
72        */
73       unsigned *sizes;
74 
75       /**
76        * Array of offsets from the start of the VGRF space in allocation
77        * units.
78        */
79       unsigned *offsets;
80 
81       /** Total number of VGRFs allocated. */
82       unsigned count;
83 
84       /** Cumulative size in allocation units. */
85       unsigned total_size;
86 
87    private:
88       unsigned capacity;
89    };
90 }
91 
92 #endif
93