mirror of
https://gitlab.freedesktop.org/mesa/mesa.git
synced 2026-05-21 06:48:09 +02:00
f4cca9d600
prep for sparse. Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io> Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/30633>
108 lines
2.7 KiB
C
108 lines
2.7 KiB
C
/*
|
|
* Copyright 2018 Alyssa Rosenzweig
|
|
* Copyright 2019 Collabora, Ltd.
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "pool.h"
|
|
#include "agx_bo.h"
|
|
#include "agx_device.h"
|
|
|
|
/* Transient command stream pooling: command stream uploads try to simply copy
|
|
* into wherever we left off. If there isn't space, we allocate a new entry
|
|
* into the pool and copy there */
|
|
|
|
#define POOL_SLAB_SIZE (256 * 1024)
|
|
|
|
static struct agx_bo *
|
|
agx_pool_alloc_backing(struct agx_pool *pool, size_t bo_sz)
|
|
{
|
|
struct agx_bo *bo =
|
|
agx_bo_create(pool->dev, bo_sz, 0, pool->create_flags, "Pool");
|
|
|
|
util_dynarray_append(&pool->bos, struct agx_bo *, bo);
|
|
pool->transient_bo = bo;
|
|
pool->transient_offset = 0;
|
|
|
|
return bo;
|
|
}
|
|
|
|
void
|
|
agx_pool_init(struct agx_pool *pool, struct agx_device *dev,
|
|
unsigned create_flags, bool prealloc)
|
|
{
|
|
memset(pool, 0, sizeof(*pool));
|
|
pool->dev = dev;
|
|
pool->create_flags = create_flags;
|
|
util_dynarray_init(&pool->bos, NULL);
|
|
|
|
if (prealloc)
|
|
agx_pool_alloc_backing(pool, POOL_SLAB_SIZE);
|
|
}
|
|
|
|
void
|
|
agx_pool_cleanup(struct agx_pool *pool)
|
|
{
|
|
util_dynarray_foreach(&pool->bos, struct agx_bo *, bo) {
|
|
agx_bo_unreference(pool->dev, *bo);
|
|
}
|
|
|
|
util_dynarray_fini(&pool->bos);
|
|
}
|
|
|
|
void
|
|
agx_pool_get_bo_handles(struct agx_pool *pool, uint32_t *handles)
|
|
{
|
|
unsigned idx = 0;
|
|
util_dynarray_foreach(&pool->bos, struct agx_bo *, bo) {
|
|
handles[idx++] = (*bo)->handle;
|
|
}
|
|
}
|
|
|
|
struct agx_ptr
|
|
agx_pool_alloc_aligned_with_bo(struct agx_pool *pool, size_t sz,
|
|
unsigned alignment, struct agx_bo **out_bo)
|
|
{
|
|
assert(alignment == util_next_power_of_two(alignment));
|
|
|
|
/* Find or create a suitable BO */
|
|
struct agx_bo *bo = pool->transient_bo;
|
|
unsigned offset = ALIGN_POT(pool->transient_offset, alignment);
|
|
|
|
/* If we don't fit, allocate a new backing */
|
|
if (unlikely(bo == NULL || (offset + sz) >= POOL_SLAB_SIZE)) {
|
|
bo = agx_pool_alloc_backing(pool,
|
|
ALIGN_POT(MAX2(POOL_SLAB_SIZE, sz), 4096));
|
|
offset = 0;
|
|
}
|
|
|
|
pool->transient_offset = offset + sz;
|
|
|
|
struct agx_ptr ret = {
|
|
.cpu = bo->map + offset,
|
|
.gpu = bo->va->addr + offset,
|
|
};
|
|
|
|
if (out_bo)
|
|
*out_bo = bo;
|
|
|
|
return ret;
|
|
}
|
|
|
|
uint64_t
|
|
agx_pool_upload(struct agx_pool *pool, const void *data, size_t sz)
|
|
{
|
|
return agx_pool_upload_aligned(pool, data, sz, util_next_power_of_two(sz));
|
|
}
|
|
|
|
uint64_t
|
|
agx_pool_upload_aligned_with_bo(struct agx_pool *pool, const void *data,
|
|
size_t sz, unsigned alignment,
|
|
struct agx_bo **bo)
|
|
{
|
|
struct agx_ptr transfer =
|
|
agx_pool_alloc_aligned_with_bo(pool, sz, alignment, bo);
|
|
memcpy(transfer.cpu, data, sz);
|
|
return transfer.gpu;
|
|
}
|