gpu: nvgpu: remove platform device on exit

Add ->remove() for undoing the ->probe() and ->late_probe() in
gk20a_platform devices, and call it when gk20a is removed.

Bug 1476801

Change-Id: Ic9b29c0a7ea4a4cae7b5a0f66774bd799eb28434
Signed-off-by: Konsta Holtta <kholtta@nvidia.com>
Reviewed-on: http://git-master/r/594443
Reviewed-by: Automatic_Commit_Validation_User
Reviewed-by: Terje Bergstrom <tbergstrom@nvidia.com>

1 files changed, 10 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/platform_gk20a_generic.c b/drivers/gpu/nvgpu/gk20a/platform_gk20a_generic.c
index d5b82fcc..ee176b8b 100644
--- a/drivers/gpu/nvgpu/gk20a/platform_gk20a_generic.c
+++ b/drivers/gpu/nvgpu/gk20a/platform_gk20a_generic.c
@@ -120,6 +120,15 @@ static int gk20a_generic_late_probe(struct platform_device *dev)
         return 0;
 }
 
+static int gk20a_generic_remove(struct platform_device *dev)
+{
+        struct gk20a_platform *platform = gk20a_get_platform(dev);
+
+        tegra_pd_remove_sd(&platform->g->pd);
+
+        return 0;
+}
+
 struct gk20a_platform gk20a_generic_platform = {
         .railgate = gk20a_generic_railgate,
         .unrailgate = gk20a_generic_unrailgate,
@@ -127,5 +136,6 @@ struct gk20a_platform gk20a_generic_platform = {
 
         .probe = gk20a_generic_probe,
         .late_probe = gk20a_generic_late_probe,
+        .remove = gk20a_generic_remove,
         .default_big_page_size  = SZ_128K,
 };