src/gallium/drivers/r600/compute_memory_pool.h

   1 /*
   2  * Permission is hereby granted, free of charge, to any person obtaining a
   3  * copy of this software and associated documentation files (the "Software"),
   4  * to deal in the Software without restriction, including without limitation
   5  * on the rights to use, copy, modify, merge, publish, distribute, sub
   6  * license, and/or sell copies of the Software, and to permit persons to whom
   7  * the Software is furnished to do so, subject to the following conditions:
   8  *
   9  * The above copyright notice and this permission notice (including the next
  10  * paragraph) shall be included in all copies or substantial portions of the
  11  * Software.
  12  *
  13  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  14  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  15  * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
  16  * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM,
  17  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  18  * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
  19  * USE OR OTHER DEALINGS IN THE SOFTWARE.
  20  *
  21  * Authors:
  22  *      Adam Rak <adam.rak@streamnovation.com>
  23  */
  24
  25 #ifndef COMPUTE_MEMORY_POOL
  26 #define COMPUTE_MEMORY_POOL
  27
  28 #include <stdlib.h>
  29
  30 #define ITEM_MAPPED_FOR_READING (1<<0)
  31 #define ITEM_MAPPED_FOR_WRITING (1<<1)
  32 #define ITEM_FOR_PROMOTING      (1<<2)
  33 #define ITEM_FOR_DEMOTING       (1<<3)
  34
  35 struct compute_memory_pool;
  36
  37 struct compute_memory_item
  38 {
  39         int64_t id; ///ID of the memory chunk
  40
  41         uint32_t status; ///Will track the status of the item
  42
  43         int64_t start_in_dw; ///Start pointer in dwords relative in the pool bo
  44         int64_t size_in_dw; ///Size of the chunk in dwords
  45
  46         struct r600_resource *real_buffer;
  47
  48         struct compute_memory_pool* pool;
  49
  50         struct compute_memory_item* prev;
  51         struct compute_memory_item* next;
  52 };
  53
  54 struct compute_memory_pool
  55 {
  56         int64_t next_id; ///For generating unique IDs for memory chunks
  57         int64_t size_in_dw; ///Size of the pool in dwords
  58
  59         struct r600_resource *bo; ///The pool buffer object resource
  60         struct compute_memory_item* item_list; ///Allocated memory chunks in the buffer,they must be ordered by "start_in_dw"
  61         struct compute_memory_item* unallocated_list; ///Unallocated memory chunks
  62         struct r600_screen *screen;
  63
  64         uint32_t *shadow; ///host copy of the pool, used for defragmentation
  65 };
  66
  67
  68 static inline int is_item_in_pool(struct compute_memory_item *item)
  69 {
  70         return item->start_in_dw != -1;
  71 }
  72
  73 struct compute_memory_pool* compute_memory_pool_new(struct r600_screen *rscreen); ///Creates a new pool
  74 void compute_memory_pool_delete(struct compute_memory_pool* pool); ///Frees all stuff in the pool and the pool struct itself too
  75
  76 int64_t compute_memory_prealloc_chunk(struct compute_memory_pool* pool, int64_t size_in_dw); ///searches for an empty space in the pool, return with the pointer to the allocatable space in the pool, returns -1 on failure
  77
  78 struct compute_memory_item* compute_memory_postalloc_chunk(struct compute_memory_pool* pool, int64_t start_in_dw); ///search for the chunk where we can link our new chunk after it
  79
  80 int compute_memory_grow_pool(struct compute_memory_pool* pool, struct pipe_context * pipe,
  81         int new_size_in_dw);
  82
  83 void compute_memory_shadow(struct compute_memory_pool* pool,
  84         struct pipe_context * pipe, int device_to_host);
  85
  86 int compute_memory_finalize_pending(struct compute_memory_pool* pool,
  87         struct pipe_context * pipe);
  88
  89 int compute_memory_promote_item(struct compute_memory_pool *pool,
  90                 struct compute_memory_item *item, struct pipe_context *pipe,
  91                 int64_t allocated);
  92
  93 void compute_memory_demote_item(struct compute_memory_pool *pool,
  94         struct compute_memory_item *item, struct pipe_context *pipe);
  95
  96 void compute_memory_free(struct compute_memory_pool* pool, int64_t id);
  97 struct compute_memory_item* compute_memory_alloc(struct compute_memory_pool* pool, int64_t size_in_dw); ///Creates pending allocations
  98
  99 void compute_memory_transfer(struct compute_memory_pool* pool,
 100         struct pipe_context * pipe, int device_to_host,
 101         struct compute_memory_item* chunk, void* data,
 102         int offset_in_chunk, int size);
 103
 104 void compute_memory_transfer_direct(struct compute_memory_pool* pool, int chunk_to_data, struct compute_memory_item* chunk, struct r600_resource* data, int offset_in_chunk, int offset_in_data, int size); ///Transfer data between chunk<->data, it is for VRAM<->GART transfers
 105
 106 #endif