1 files changed, 39 insertions, 0 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/psp_v11_0.c b/drivers/gpu/drm/amd/amdgpu/psp_v11_0.c
index 7b248915489a..f56f8e376b62 100644
--- a/drivers/gpu/drm/amd/amdgpu/psp_v11_0.c
+++ b/drivers/gpu/drm/amd/amdgpu/psp_v11_0.c
@@ -570,6 +570,45 @@ static int psp_v11_0_mode1_reset(struct psp_context *psp)
 static int psp_v11_0_xgmi_get_topology_info(struct psp_context *psp,
         int number_devices, struct psp_xgmi_topology_info *topology)
 {
+        struct ta_xgmi_shared_memory *xgmi_cmd;
+        struct ta_xgmi_cmd_get_topology_info_input *topology_info_input;
+        struct ta_xgmi_cmd_get_topology_info_output *topology_info_output;
+        int i;
+        int ret;
+
+        if (!topology || topology->num_nodes > TA_XGMI__MAX_CONNECTED_NODES)
+                return -EINVAL;
+
+        xgmi_cmd = (struct ta_xgmi_shared_memory*)psp->xgmi_context.xgmi_shared_buf;
+        memset(xgmi_cmd, 0, sizeof(struct ta_xgmi_shared_memory));
+
+        /* Fill in the shared memory with topology information as input */
+        topology_info_input = &xgmi_cmd->xgmi_in_message.get_topology_info;
+        xgmi_cmd->cmd_id = TA_COMMAND_XGMI__GET_GET_TOPOLOGY_INFO;
+        topology_info_input->num_nodes = number_devices;
+
+        for (i = 0; i < topology_info_input->num_nodes; i++) {
+                topology_info_input->nodes[i].node_id = topology->nodes[i].node_id;
+                topology_info_input->nodes[i].num_hops = topology->nodes[i].num_hops;
+                topology_info_input->nodes[i].is_sharing_enabled = topology->nodes[i].is_sharing_enabled;
+                topology_info_input->nodes[i].sdma_engine = topology->nodes[i].sdma_engine;
+        }
+
+        /* Invoke xgmi ta to get the topology information */
+        ret = psp_xgmi_invoke(psp, TA_COMMAND_XGMI__GET_GET_TOPOLOGY_INFO);
+        if (ret)
+                return ret;
+
+        /* Read the output topology information from the shared memory */
+        topology_info_output = &xgmi_cmd->xgmi_out_message.get_topology_info;
+        topology->num_nodes = xgmi_cmd->xgmi_out_message.get_topology_info.num_nodes;
+        for (i = 0; i < topology->num_nodes; i++) {
+                topology->nodes[i].node_id = topology_info_output->nodes[i].node_id;
+                topology->nodes[i].num_hops = topology_info_output->nodes[i].num_hops;
+                topology->nodes[i].is_sharing_enabled = topology_info_output->nodes[i].is_sharing_enabled;
+                topology->nodes[i].sdma_engine = topology_info_output->nodes[i].sdma_engine;
+        }
+
         return 0;
 }