gpu: nvgpu: remove nvlink on driver removal

Unregister nvlink and nvlink device when gpu is
getting removed. Without this next modprobe of
nvgpu results in nvlink registration failure.

Bug 1987855

Change-Id: I785e707d1fa90f45a3ff0e9790f3f02fa15510d4
Signed-off-by: Nitin Kumbhar <nkumbhar@nvidia.com>
Reviewed-on: https://git-master.nvidia.com/r/1735986
Reviewed-by: mobile promotions <svcmobile_promotions@nvidia.com>
Tested-by: mobile promotions <svcmobile_promotions@nvidia.com>

1 files changed, 30 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/common/nvlink.c b/drivers/gpu/nvgpu/common/nvlink.c
index 25c1b10b..1a6ee853 100644
--- a/drivers/gpu/nvgpu/common/nvlink.c
+++ b/drivers/gpu/nvgpu/common/nvlink.c
@@ -503,3 +503,33 @@ free_ndev:
         return -ENODEV;
 #endif
 }
+
+int nvgpu_nvlink_remove(struct gk20a *g)
+{
+#ifdef CONFIG_TEGRA_NVLINK
+        struct nvlink_device *ndev;
+        int err;
+
+        ndev = g->nvlink.priv;
+
+        __nvgpu_set_enabled(g, NVGPU_SUPPORT_NVLINK, false);
+
+        err = nvlink_unregister_link(&ndev->link);
+        if (err) {
+                nvgpu_err(g, "failed on nvlink link unregistration");
+                return err;
+        }
+
+        err = nvlink_unregister_device(ndev);
+        if (err) {
+                nvgpu_err(g, "failed on nvlink device unregistration");
+                return err;
+        }
+
+        nvgpu_kfree(g, ndev);
+
+        return 0;
+#else
+        return -ENODEV;
+#endif
+}