1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c b/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c
index 3bcbdfd9..34351f93 100644
--- a/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c
@@ -258,7 +258,6 @@ long gk20a_ctrl_dev_ioctl(struct file *filp, unsigned int cmd, unsigned long arg
         struct zbc_entry *zbc_val;
         struct zbc_query_params *zbc_tbl;
         int i, err = 0;
-        struct gk20a_platform *platform = platform_get_drvdata(dev);
 
         gk20a_dbg_fn("");
 
@@ -320,7 +319,8 @@ long gk20a_ctrl_dev_ioctl(struct file *filp, unsigned int cmd, unsigned long arg
         case NVGPU_GPU_IOCTL_ZBC_SET_TABLE:
                 set_table_args = (struct nvgpu_gpu_zbc_set_table_args *)buf;
 
-                if (platform->virtual_dev)
+                /* not supported for vgpu */
+                if (gk20a_gpu_is_virtual(dev))
                         return -ENOMEM;
 
                 zbc_val = kzalloc(sizeof(struct zbc_entry), GFP_KERNEL);