1 files changed, 18 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c b/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c
index 4dee3faf..7b617a03 100644
--- a/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c
@@ -243,6 +243,20 @@ static int gk20a_ctrl_get_tpc_masks(struct gk20a *g,
         return err;
 }
 
+static int nvgpu_gpu_ioctl_l2_fb_ops(struct gk20a *g,
+                struct nvgpu_gpu_l2_fb_args *args)
+{
+        int err = 0;
+
+        if (args->l2_flush)
+                g->ops.mm.l2_flush(g, args->l2_invalidate ? true : false);
+
+        if (args->fb_flush)
+                g->ops.mm.fb_flush(g);
+
+        return err;
+}
+
 long gk20a_ctrl_dev_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
 {
         struct platform_device *dev = filp->private_data;
@@ -423,6 +437,10 @@ long gk20a_ctrl_dev_ioctl(struct file *filp, unsigned int cmd, unsigned long arg
                 err = gk20a_channel_open_ioctl(g,
                         (struct nvgpu_channel_open_args *)buf);
                 break;
+        case NVGPU_GPU_IOCTL_FLUSH_L2:
+                err = nvgpu_gpu_ioctl_l2_fb_ops(g,
+                           (struct nvgpu_gpu_l2_fb_args *)buf);
+                break;
         default:
                 dev_dbg(dev_from_gk20a(g), "unrecognized gpu ioctl cmd: 0x%x", cmd);
                 err = -ENOTTY;