diff options
Diffstat (limited to 'drivers/gpu')
-rw-r--r-- | drivers/gpu/nvgpu/gk20a/gr_gk20a.c | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c index 94f66e83..54c96680 100644 --- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c +++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c | |||
@@ -670,6 +670,12 @@ void gr_gk20a_ctx_patch_write_end(struct gk20a *g, | |||
670 | struct channel_ctx_gk20a *ch_ctx) | 670 | struct channel_ctx_gk20a *ch_ctx) |
671 | { | 671 | { |
672 | gk20a_mem_end(g, &ch_ctx->patch_ctx.mem); | 672 | gk20a_mem_end(g, &ch_ctx->patch_ctx.mem); |
673 | /* Write context count to context image if it is mapped */ | ||
674 | if (ch_ctx->gr_ctx->mem.cpu_va) { | ||
675 | gk20a_mem_wr(g, &ch_ctx->gr_ctx->mem, | ||
676 | ctxsw_prog_main_image_patch_count_o(), | ||
677 | ch_ctx->patch_ctx.data_count); | ||
678 | } | ||
673 | } | 679 | } |
674 | 680 | ||
675 | void gr_gk20a_ctx_patch_write(struct gk20a *g, | 681 | void gr_gk20a_ctx_patch_write(struct gk20a *g, |