mirror of
https://gitlab.freedesktop.org/mesa/mesa.git
synced 2025-12-25 13:00:09 +01:00
radeonsi/compute: Stop leaking the input buffer
We were leaking the input buffer used for kernel arguments and since we were allocating it using si_upload_const_buffer() we were leaking 1 MB per kernel invocation. CC: "10.2" <mesa-stable@lists.freedesktop.org>
This commit is contained in:
parent
38fccc37c1
commit
a15088338e
1 changed files with 14 additions and 7 deletions
|
|
@ -48,6 +48,7 @@ struct si_pipe_compute {
|
|||
struct si_pipe_shader *kernels;
|
||||
unsigned num_user_sgprs;
|
||||
|
||||
struct r600_resource *input_buffer;
|
||||
struct pipe_resource *global_buffers[MAX_GLOBAL_BUFFERS];
|
||||
|
||||
LLVMContextRef llvm_ctx;
|
||||
|
|
@ -85,6 +86,9 @@ static void *si_create_compute_state(
|
|||
LLVMDisposeModule(mod);
|
||||
}
|
||||
|
||||
program->input_buffer = si_resource_create_custom(sctx->b.b.screen,
|
||||
PIPE_USAGE_IMMUTABLE, program->input_size);
|
||||
|
||||
return program;
|
||||
}
|
||||
|
||||
|
|
@ -167,7 +171,7 @@ static void si_launch_grid(
|
|||
struct si_context *sctx = (struct si_context*)ctx;
|
||||
struct si_pipe_compute *program = sctx->cs_shader_state.program;
|
||||
struct si_pm4_state *pm4 = CALLOC_STRUCT(si_pm4_state);
|
||||
struct r600_resource *kernel_args_buffer = NULL;
|
||||
struct r600_resource *input_buffer = program->input_buffer;
|
||||
unsigned kernel_args_size;
|
||||
unsigned num_work_size_bytes = 36;
|
||||
uint32_t kernel_args_offset = 0;
|
||||
|
|
@ -199,7 +203,8 @@ static void si_launch_grid(
|
|||
/* The extra num_work_size_bytes are for work group / work item size information */
|
||||
kernel_args_size = program->input_size + num_work_size_bytes + 8 /* For scratch va */;
|
||||
|
||||
kernel_args = MALLOC(kernel_args_size);
|
||||
kernel_args = sctx->b.ws->buffer_map(input_buffer->cs_buf,
|
||||
sctx->b.rings.gfx.cs, PIPE_TRANSFER_WRITE);
|
||||
for (i = 0; i < 3; i++) {
|
||||
kernel_args[i] = grid_layout[i];
|
||||
kernel_args[i + 3] = grid_layout[i] * block_layout[i];
|
||||
|
|
@ -235,12 +240,13 @@ static void si_launch_grid(
|
|||
kernel_args[i]);
|
||||
}
|
||||
|
||||
si_upload_const_buffer(sctx, &kernel_args_buffer, (uint8_t*)kernel_args,
|
||||
kernel_args_size, &kernel_args_offset);
|
||||
kernel_args_va = kernel_args_buffer->gpu_address;
|
||||
sctx->b.ws->buffer_unmap(input_buffer->cs_buf);
|
||||
|
||||
kernel_args_va = input_buffer->gpu_address;
|
||||
kernel_args_va += kernel_args_offset;
|
||||
|
||||
si_pm4_add_bo(pm4, kernel_args_buffer, RADEON_USAGE_READ, RADEON_PRIO_SHADER_DATA);
|
||||
si_pm4_add_bo(pm4, input_buffer, RADEON_USAGE_READ,
|
||||
RADEON_PRIO_SHADER_DATA);
|
||||
|
||||
si_pm4_set_reg(pm4, R_00B900_COMPUTE_USER_DATA_0, kernel_args_va);
|
||||
si_pm4_set_reg(pm4, R_00B900_COMPUTE_USER_DATA_0 + 4, S_008F04_BASE_ADDRESS_HI (kernel_args_va >> 32) | S_008F04_STRIDE(0));
|
||||
|
|
@ -372,7 +378,6 @@ static void si_launch_grid(
|
|||
}
|
||||
#endif
|
||||
|
||||
FREE(kernel_args);
|
||||
si_pm4_free_state(sctx, pm4, ~0);
|
||||
}
|
||||
|
||||
|
|
@ -396,6 +401,8 @@ static void si_delete_compute_state(struct pipe_context *ctx, void* state){
|
|||
if (program->llvm_ctx){
|
||||
LLVMContextDispose(program->llvm_ctx);
|
||||
}
|
||||
pipe_resource_reference(
|
||||
(struct pipe_resource **)&program->input_buffer, NULL);
|
||||
|
||||
//And then free the program itself.
|
||||
FREE(program);
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue