1 files changed, 11 insertions, 5 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index afe1447e..ca6a8ae0 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -713,15 +713,21 @@ enum {
         KEPLER_CHANNEL_GPFIFO_C   = 0xA26F,
 };
 
+static inline bool gk20a_gpu_is_virtual(struct platform_device *dev)
+{
+        struct gk20a_platform *platform = gk20a_get_platform(dev);
+
+        return platform->virtual_dev;
+}
+
 static inline int support_gk20a_pmu(struct platform_device *dev)
 {
         if (IS_ENABLED(CONFIG_GK20A_PMU)) {
-                struct gk20a_platform *platform = gk20a_get_platform(dev);
+                /* gPMU is not supported for vgpu */
+                return !gk20a_gpu_is_virtual(dev);
+        }
 
-                /* we have not supported GPU PMU for virtualization now */
-                return !platform->virtual_dev;
-        } else
-                return 0;
+        return 0;
 }
 
 void gk20a_create_sysfs(struct platform_device *dev);