xref: /aosp_15_r20/external/mesa3d/src/intel/compiler/elk/elk_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 ELK_IR_ALLOCATOR_H
26 #define ELK_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 elk {
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 
45       simple_allocator(const simple_allocator &) = delete;
46 
~simple_allocator()47       ~simple_allocator()
48       {
49          free(offsets);
50          free(sizes);
51       }
52 
53       simple_allocator & operator=(const simple_allocator &) = delete;
54 
55       unsigned
allocate(unsigned size)56       allocate(unsigned size)
57       {
58          assert(size > 0);
59          if (capacity <= count) {
60             capacity = MAX2(16, capacity * 2);
61             sizes = (unsigned *)realloc(sizes, capacity * sizeof(unsigned));
62             offsets = (unsigned *)realloc(offsets, capacity * sizeof(unsigned));
63          }
64 
65          sizes[count] = size;
66          offsets[count] = total_size;
67          total_size += size;
68 
69          return count++;
70       }
71 
72       /**
73        * Array of sizes for each allocation.  The allocation unit is up to the
74        * back-end, but it's expected to be one scalar value in the FS back-end
75        * and one vec4 in the VEC4 back-end.
76        */
77       unsigned *sizes;
78 
79       /**
80        * Array of offsets from the start of the VGRF space in allocation
81        * units.
82        */
83       unsigned *offsets;
84 
85       /** Total number of VGRFs allocated. */
86       unsigned count;
87 
88       /** Cumulative size in allocation units. */
89       unsigned total_size;
90 
91    private:
92       unsigned capacity;
93    };
94 }
95 
96 #endif
97