1 files changed, 8 insertions, 6 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 38322c87..979b4cb0 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -687,14 +687,16 @@ enum {
         KEPLER_DMA_COPY_A         = 0xA0B5, /*not sure about this one*/
 };
 
-#if defined(CONFIG_GK20A_PMU)
-static inline int support_gk20a_pmu(void)
+static inline int support_gk20a_pmu(struct platform_device *dev)
 {
-        return 1;
+        if (IS_ENABLED(CONFIG_GK20A_PMU)) {
+                struct gk20a_platform *platform = gk20a_get_platform(dev);
+
+                /* we have not supported GPU PMU for virtualization now */
+                return !platform->virtual_dev;
+        } else
+                return 0;
 }
-#else
-static inline int support_gk20a_pmu(void){return 0;}
-#endif
 
 void gk20a_create_sysfs(struct platform_device *dev);