1 files changed, 2 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gp106/clk_arb_gp106.c b/drivers/gpu/nvgpu/gp106/clk_arb_gp106.c
index b4d1afbc..4a907521 100644
--- a/drivers/gpu/nvgpu/gp106/clk_arb_gp106.c
+++ b/drivers/gpu/nvgpu/gp106/clk_arb_gp106.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2016, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2016-2017, NVIDIA CORPORATION.  All rights reserved.
  *
  * This program is free software; you can redistribute it and/or modify it
  * under the terms and conditions of the GNU General Public License,
@@ -102,4 +102,5 @@ void gp106_init_clk_arb_ops(struct gpu_ops *gops)
         gops->clk_arb.get_arbiter_clk_domains = gp106_get_arbiter_clk_domains;
         gops->clk_arb.get_arbiter_clk_range = gp106_get_arbiter_clk_range;
         gops->clk_arb.get_arbiter_clk_default = gp106_get_arbiter_clk_default;
+        gops->clk_arb.get_current_pstate = nvgpu_clk_arb_get_current_pstate;
 }