X-Git-Url: http://plrg.eecs.uci.edu/git/?p=firefly-linux-kernel-4.4.55.git;a=blobdiff_plain;f=drivers%2Fgpu%2Farm%2Fmali400%2Fmali%2Fplatform%2Frk30%2Frk3066.c;h=f7d264459da1da9ad51c1ec5ec8f52f99a69b3f2;hp=e8c607e59af966838440f375af29adb0ddb93bca;hb=3c562205635af8e12959f8ccf3e385f0f17f0537;hpb=c3b4c5336b6eff35ff6f0a44761fd37a4243fbb4

diff --git a/drivers/gpu/arm/mali400/mali/platform/rk30/rk3066.c b/drivers/gpu/arm/mali400/mali/platform/rk30/rk3066.c
index e8c607e59af9..f7d264459da1 100644
--- a/drivers/gpu/arm/mali400/mali/platform/rk30/rk3066.c
+++ b/drivers/gpu/arm/mali400/mali/platform/rk30/rk3066.c
@@ -27,6 +27,7 @@
 #include <linux/platform_device.h>
 #include <linux/version.h>
 #include <linux/pm.h>
+#include <linux/of.h>
 #ifdef CONFIG_PM_RUNTIME
 #include <linux/pm_runtime.h>
 #endif
@@ -224,18 +225,14 @@ int mali_platform_device_init(struct platform_device *pdev)
 
 	D("mali_platform_device_register() called\n");
 
-	/*
-	if (cpu_is_rk312x())
-		num_pp_cores = 2;
-	else if (cpu_is_rk3036())
-	*/
+	if (of_machine_is_compatible("rockchip,rk3036"))
 		num_pp_cores = 1;
-	/*
-	else if (cpu_is_rk3188())
-		num_pp_cores = 4;
-	else if (cpu_is_rk3228())
+	else if (of_machine_is_compatible("rockchip,rk3228h"))
+		num_pp_cores = 2;
+	else if (of_machine_is_compatible("rockchip,rk3328h"))
+		num_pp_cores = 2;
+	else
 		num_pp_cores = 2;
-	*/
 
 	D("to add config.");
 	mali_platform_device_add_config(pdev);