diff options
Diffstat (limited to 'drivers/gpu/nvgpu/gp10b')
-rw-r--r-- | drivers/gpu/nvgpu/gp10b/gr_gp10b.c | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c index 03462d5f..33a52db5 100644 --- a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c +++ b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c | |||
@@ -475,7 +475,7 @@ static int gr_gp10b_alloc_gr_ctx(struct gk20a *g, | |||
475 | if (err) | 475 | if (err) |
476 | return err; | 476 | return err; |
477 | 477 | ||
478 | if (flags == NVGPU_GR_PREEMPTION_MODE_GFXP) { | 478 | if (flags & NVGPU_ALLOC_OBJ_FLAGS_GFXP) { |
479 | u32 spill_size = | 479 | u32 spill_size = |
480 | gr_gpc0_swdx_rm_spill_buffer_size_256b_default_v(); | 480 | gr_gpc0_swdx_rm_spill_buffer_size_256b_default_v(); |
481 | u32 pagepool_size = g->ops.gr.pagepool_default_size(g) * | 481 | u32 pagepool_size = g->ops.gr.pagepool_default_size(g) * |
@@ -528,7 +528,7 @@ static int gr_gp10b_alloc_gr_ctx(struct gk20a *g, | |||
528 | } | 528 | } |
529 | 529 | ||
530 | if (class == PASCAL_COMPUTE_A) { | 530 | if (class == PASCAL_COMPUTE_A) { |
531 | if (flags == NVGPU_GR_PREEMPTION_MODE_CILP) | 531 | if (flags & NVGPU_ALLOC_OBJ_FLAGS_CILP) |
532 | (*gr_ctx)->preempt_mode = NVGPU_GR_PREEMPTION_MODE_CILP; | 532 | (*gr_ctx)->preempt_mode = NVGPU_GR_PREEMPTION_MODE_CILP; |
533 | else | 533 | else |
534 | (*gr_ctx)->preempt_mode = NVGPU_GR_PREEMPTION_MODE_CTA; | 534 | (*gr_ctx)->preempt_mode = NVGPU_GR_PREEMPTION_MODE_CTA; |