OpenHarmony/third_party_mesa3d
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

anv/allocator: Simplify anv_scratch_pool

Browse Source 
The previous implementation was being overly clever and using the
anv_bo::size field as its mutex.  Scratch pool allocations don't happen
often, will happen at most a fixed number of times, and never happen in the
critical path (they only happen in shader compilation).  We can make this
much simpler by just using the device mutex.  This also means that we can
start using anv_bo_init_new directly on the bo and avoid setting fields
one-at-a-time.

Signed-off-by: Jason Ekstrand <jason@jlekstrand.net>
Reviewed-by: Kristian H. Kristensen <hoegsberg@google.com>
Cc: "13.0" <mesa-stable@lists.freedesktop.org>
This commit is contained in:
Jason Ekstrand
2016-11-01 13:10:11 -07:00
parent 6283b6d56a
commit bd0f8d5070
2 changed files with 56 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
src/intel/vulkan/anv_allocator.c
View File

@@ -888,9 +888,9 @@ anv_scratch_pool_finish(struct anv_device *device, struct anv_scratch_pool *pool
{
for (unsigned s = 0; s < MESA_SHADER_STAGES; s++) {
for (unsigned i = 0; i < 16; i++) {
struct anv_bo *bo = &pool->bos[i][s];
if (bo->size > 0)
anv_gem_close(device, bo->gem_handle);
struct anv_scratch_bo *bo = &pool->bos[i][s];
if (bo->exists > 0)
anv_gem_close(device, bo->bo.gem_handle);
}
}
}
@@ -905,20 +905,17 @@ anv_scratch_pool_alloc(struct anv_device *device, struct anv_scratch_pool *pool,
unsigned scratch_size_log2 = ffs(per_thread_scratch / 2048);
assert(scratch_size_log2 < 16);
struct anv_bo *bo = &pool->bos[scratch_size_log2][stage];
struct anv_scratch_bo *bo = &pool->bos[scratch_size_log2][stage];
/* From now on, we go into a critical section. In order to remain
* thread-safe, we use the bo size as a lock. A value of 0 means we don't
* have a valid BO yet. A value of 1 means locked. A value greater than 1
* means we have a bo of the given size.
*/
/* We can use "exists" to shortcut and ignore the critical section */
if (bo->exists)
return &bo->bo;
if (bo->size > 1)
return bo;
pthread_mutex_lock(&device->mutex);
uint64_t size = __sync_val_compare_and_swap(&bo->size, 0, 1);
if (size == 0) {
/* We own the lock. Allocate a buffer */
__sync_synchronize();
if (bo->exists)
return &bo->bo;
const struct anv_physical_device *physical_device =
&device->instance->physicalDevice;
@@ -952,23 +949,15 @@ anv_scratch_pool_alloc(struct anv_device *device, struct anv_scratch_pool *pool,
[MESA_SHADER_COMPUTE] = scratch_ids_per_subslice * subslices,
};
size = per_thread_scratch * max_threads[stage];
uint32_t size = per_thread_scratch * max_threads[stage];
struct anv_bo new_bo;
anv_bo_init_new(&new_bo, device, size);
anv_bo_init_new(&bo->bo, device, size);
bo->gem_handle = new_bo.gem_handle;
/* Set the size last because we use it as a lock */
/* Set the exists last because it may be read by other threads */
__sync_synchronize();
bo->size = size;
bo->exists = true;
futex_wake((uint32_t *)&bo->size, INT_MAX);
} else {
/* Someone else got here first */
while (bo->size == 1)
futex_wait((uint32_t *)&bo->size, 1);
}
pthread_mutex_unlock(&device->mutex);
return bo;
return &bo->bo;
}

7
src/intel/vulkan/anv_private.h
View File

@@ -453,9 +453,14 @@ VkResult anv_bo_pool_alloc(struct anv_bo_pool *pool, struct anv_bo *bo,
uint32_t size);
void anv_bo_pool_free(struct anv_bo_pool *pool, const struct anv_bo *bo);
struct anv_scratch_bo {
bool exists;
struct anv_bo bo;
};
struct anv_scratch_pool {
/* Indexed by Per-Thread Scratch Space number (the hardware value) and stage */
struct anv_bo bos[16][MESA_SHADER_STAGES];
struct anv_scratch_bo bos[16][MESA_SHADER_STAGES];
};
void anv_scratch_pool_init(struct anv_device *device,