nvlink: add shim driver shutdown IOCTLs

For the shim driver mode, RM needs to perform a graceful shutdown of
NVLINK during RM unload. Export the following T19x NVLINK endpoint
IOCTLs for shim driver NVLINK shutdown:
   - INTERFACE_DISABLE IOCTL: Disables the NVLINK aperture
   - FINALIZE_SHUTDOWN IOCTL: Does shutdown related SW cleanup

Bug 2113729
Jira NVLINK-173

Change-Id: I60e3f5fffd0b1e6cc87476b047ef1b761b5174d1
Signed-off-by: Adeel Raza <araza@nvidia.com>
Reviewed-on: https://git-master.nvidia.com/r/1714177
Reviewed-by: Automatic_Commit_Validation_User
Reviewed-by: svc-mobile-coverity <svc-mobile-coverity@nvidia.com>
GVS: Gerrit_Virtual_Submit
Reviewed-by: Petlozu Pravareshwar <petlozup@nvidia.com>
Reviewed-by: mobile promotions <svcmobile_promotions@nvidia.com>
Tested-by: mobile promotions <svcmobile_promotions@nvidia.com>

2 files changed, 9 insertions, 0 deletions

diff --git a/include/linux/platform/tegra/tegra-nvlink.h b/include/linux/platform/tegra/tegra-nvlink.h
index 66cec6115..09798b293 100644
--- a/include/linux/platform/tegra/tegra-nvlink.h
+++ b/include/linux/platform/tegra/tegra-nvlink.h
@@ -303,6 +303,7 @@ int nvlink_register_link(struct nvlink_link* link);
 int nvlink_unregister_device(struct nvlink_device* device);
 int nvlink_unregister_link(struct nvlink_link* link);
 int nvlink_get_init_state(struct nvlink_device *ndev, enum init_state *state);
+int nvlink_set_init_state(struct nvlink_device *ndev, enum init_state state);
 int nvlink_enumerate(struct nvlink_device *ndev);
 int nvlink_transition_intranode_conn_off_to_safe(struct nvlink_device *ndev);
 int nvlink_train_intranode_conn_safe_to_hs(struct nvlink_device *ndev);
diff --git a/include/uapi/linux/tegra-nvlink-uapi.h b/include/uapi/linux/tegra-nvlink-uapi.h
index 099ecaeb0..7ac8a81aa 100644
--- a/include/uapi/linux/tegra-nvlink-uapi.h
+++ b/include/uapi/linux/tegra-nvlink-uapi.h
@@ -518,6 +518,8 @@ enum tnvlink_ioctl_num {
         TNVLINK_IOCTL_READ_DISCOVERY_TOKEN,
         TNVLINK_IOCTL_GET_LOCAL_PCI_INFO,
         TNVLINK_IOCTL_SET_TOPOLOGY_INFO,
+        TNVLINK_IOCTL_INTERFACE_DISABLE,
+        TNVLINK_IOCTL_FINALIZE_SHUTDOWN,
         TNVLINK_IOCTL_NUM_IOCTLS
 };
 
@@ -622,5 +624,11 @@ enum tnvlink_ioctl_num {
                         _IOW(TEGRA_NVLINK_IOC_MAGIC,                    \
                                 TNVLINK_IOCTL_SET_TOPOLOGY_INFO,        \
                                 struct tegra_nvlink_set_topology_info)
+#define TEGRA_CTRL_CMD_NVLINK_INTERFACE_DISABLE                         \
+                        _IO(TEGRA_NVLINK_IOC_MAGIC,                     \
+                                TNVLINK_IOCTL_INTERFACE_DISABLE)
+#define TEGRA_CTRL_CMD_NVLINK_FINALIZE_SHUTDOWN                         \
+                        _IO(TEGRA_NVLINK_IOC_MAGIC,                     \
+                                TNVLINK_IOCTL_FINALIZE_SHUTDOWN)
 
 #endif /* TEGRA_NVLINK_UAPI_H */