drm/amdgpu/psp: add helper function to invoke xgmi ta per ta cmd_id

psp_xgmi_invoke is the helper function to issue ta cmd to firmware

Signed-off-by: Hawking Zhang <Hawking.Zhang@amd.com>
Acked-by: Alex Deucher <alexander.deucher@amd.com>
Reviewed-by: Huang Rui <ray.huang@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>

1 files changed, 36 insertions, 0 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_psp.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_psp.c
index a639bedb0ad3..07f9fcb59c0a 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_psp.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_psp.c
@@ -369,6 +369,42 @@ static int psp_xgmi_unload(struct psp_context *psp)
         return ret;
 }
 
+static void psp_prep_xgmi_ta_invoke_cmd_buf(struct psp_gfx_cmd_resp *cmd,
+                                            uint32_t ta_cmd_id,
+                                            uint32_t xgmi_session_id)
+{
+        cmd->cmd_id = GFX_CMD_ID_INVOKE_CMD;
+        cmd->cmd.cmd_invoke_cmd.session_id = xgmi_session_id;
+        cmd->cmd.cmd_invoke_cmd.ta_cmd_id = ta_cmd_id;
+        /* Note: cmd_invoke_cmd.buf is not used for now */
+}
+
+int psp_xgmi_invoke(struct psp_context *psp, uint32_t ta_cmd_id)
+{
+        int ret;
+        struct psp_gfx_cmd_resp *cmd;
+
+        /*
+         * TODO: bypass the loading in sriov for now
+        */
+        if (amdgpu_sriov_vf(psp->adev))
+                return 0;
+
+        cmd = kzalloc(sizeof(struct psp_gfx_cmd_resp), GFP_KERNEL);
+        if (!cmd)
+                return -ENOMEM;
+
+        psp_prep_xgmi_ta_invoke_cmd_buf(cmd, ta_cmd_id,
+                                        psp->xgmi_context.session_id);
+
+        ret = psp_cmd_submit_buf(psp, NULL, cmd,
+                                 psp->fence_buf_mc_addr);
+
+        kfree(cmd);
+
+        return ret;
+}
+
 static int psp_hw_start(struct psp_context *psp)
 {
         struct amdgpu_device *adev = psp->adev;