1 files changed, 81 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/platform_gk20a_generic.c b/drivers/gpu/nvgpu/gk20a/platform_gk20a_generic.c
index 7b750df6..5000db7f 100644
--- a/drivers/gpu/nvgpu/gk20a/platform_gk20a_generic.c
+++ b/drivers/gpu/nvgpu/gk20a/platform_gk20a_generic.c
@@ -18,18 +18,97 @@
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+#include <linux/tegra-powergate.h>
+#include <linux/tegra_pm_domains.h>
+#include <linux/clk.h>
+
 #include "platform_gk20a.h"
+#include "hal_gk20a.h"
+#include "gk20a.h"
+
+/*
+ * gk20a_generic_railgate()
+ *
+ * Gate (disable) gk20a power rail
+ */
+
+static int gk20a_generic_railgate(struct platform_device *pdev)
+{
+        if (tegra_powergate_is_powered(TEGRA_POWERGATE_GPU))
+                tegra_powergate_partition(TEGRA_POWERGATE_GPU);
+        return 0;
+}
+
+/*
+ * gk20a_generic_unrailgate()
+ *
+ * Ungate (enable) gk20a power rail
+ */
+
+static int gk20a_generic_unrailgate(struct platform_device *pdev)
+{
+        tegra_unpowergate_partition(TEGRA_POWERGATE_GPU);
+        return 0;
+}
+
+/*
+ * gk20a_generic_get_clocks()
+ *
+ * This function finds clocks in tegra platform and populates
+ * the clock information to gk20a platform data.
+ */
+
+static int gk20a_generic_get_clocks(struct platform_device *pdev)
+{
+        struct gk20a_platform *platform = platform_get_drvdata(pdev);
+
+        platform->clk[0] = clk_get_sys("tegra_gk20a.0", "PLLG_ref");
+        platform->clk[1] = clk_get_sys("tegra_gk20a.0", "pwr");
+        platform->clk[2] = clk_get_sys("tegra_gk20a.0", "emc");
+        platform->num_clks = 3;
+
+        if (IS_ERR(platform->clk[0]) ||
+            IS_ERR(platform->clk[1]) ||
+            IS_ERR(platform->clk[2]))
+                goto err_get_clock;
+
+        clk_set_rate(platform->clk[0], UINT_MAX);
+        clk_set_rate(platform->clk[1], 204000000);
+        clk_set_rate(platform->clk[2], UINT_MAX);
+
+        return 0;
+
+err_get_clock:
+        if (!IS_ERR_OR_NULL(platform->clk[0]))
+                clk_put(platform->clk[0]);
+        if (!IS_ERR_OR_NULL(platform->clk[1]))
+                clk_put(platform->clk[1]);
+        if (!IS_ERR_OR_NULL(platform->clk[2]))
+                clk_put(platform->clk[2]);
+        return -ENODEV;
+}
 
 static int gk20a_generic_probe(struct platform_device *dev)
 {
+        gk20a_generic_get_clocks(dev);
+
+        return 0;
+}
+
+static int gk20a_generic_late_probe(struct platform_device *dev)
+{
         struct gk20a_platform *platform = gk20a_get_platform(dev);
 
-        /* TODO: Initialize clocks and power */
-        (void)platform;
+        /* Make gk20a power domain a subdomain of mc */
+        tegra_pd_add_sd(&platform->g->pd);
 
         return 0;
 }
 
 struct gk20a_platform gk20a_generic_platform = {
+        .railgate = gk20a_generic_railgate,
+        .unrailgate = gk20a_generic_unrailgate,
+
         .probe = gk20a_generic_probe,
+        .late_probe = gk20a_generic_late_probe,
 };