2 files changed, 2 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/debug_pmu.c b/drivers/gpu/nvgpu/common/linux/debug_pmu.c
index a8a8870e..f4ed992d 100644
--- a/drivers/gpu/nvgpu/common/linux/debug_pmu.c
+++ b/drivers/gpu/nvgpu/common/linux/debug_pmu.c
@@ -27,7 +27,7 @@ static int lpwr_debug_show(struct seq_file *s, void *data)
         if (g->ops.pmu.pmu_pg_engines_feature_list &&
                 g->ops.pmu.pmu_pg_engines_feature_list(g,
                 PMU_PG_ELPG_ENGINE_ID_GRAPHICS) !=
-                PMU_PG_FEATURE_GR_POWER_GATING_ENABLED) {
+                NVGPU_PMU_GR_FEATURE_MASK_POWER_GATING) {
                 seq_printf(s, "PSTATE: %u\n"
                         "RPPG Enabled: %u\n"
                         "RPPG ref count: %u\n"
diff --git a/drivers/gpu/nvgpu/common/linux/sysfs.c b/drivers/gpu/nvgpu/common/linux/sysfs.c
index 86f1877d..afa08fc4 100644
--- a/drivers/gpu/nvgpu/common/linux/sysfs.c
+++ b/drivers/gpu/nvgpu/common/linux/sysfs.c
@@ -477,7 +477,7 @@ static ssize_t elpg_enable_store(struct device *dev,
                         if (g->ops.pmu.pmu_pg_engines_feature_list &&
                                 g->ops.pmu.pmu_pg_engines_feature_list(g,
                                 PMU_PG_ELPG_ENGINE_ID_GRAPHICS) !=
-                                PMU_PG_FEATURE_GR_POWER_GATING_ENABLED) {
+                                NVGPU_PMU_GR_FEATURE_MASK_POWER_GATING) {
                                 nvgpu_pmu_pg_global_enable(g, false);
                                 g->elpg_enabled = false;
                         } else {