189 	psp_ta_free_shared_buf(&psp->xgmi_context.context.mem_context);  in psp_free_shared_bufs()
192 	psp_ta_free_shared_buf(&psp->ras_context.context.mem_context);  in psp_free_shared_bufs()
195 	psp_ta_free_shared_buf(&psp->hdcp_context.context.mem_context);  in psp_free_shared_bufs()
198 	psp_ta_free_shared_buf(&psp->dtm_context.context.mem_context);  in psp_free_shared_bufs()
201 	psp_ta_free_shared_buf(&psp->rap_context.context.mem_context);  in psp_free_shared_bufs()
204 	psp_ta_free_shared_buf(&psp->securedisplay_context.context.mem_context);  in psp_free_shared_bufs()
968 	psp->asd_context.mem_context.shared_mc_addr  = 0;  in psp_asd_initialize()
969 	psp->asd_context.mem_context.shared_mem_size = PSP_ASD_SHARED_MEM_SIZE;  in psp_asd_initialize()
1056 		lower_32_bits(context->mem_context.shared_mc_addr);  in psp_prep_ta_load_cmd_buf()
1058 		upper_32_bits(context->mem_context.shared_mc_addr);  in psp_prep_ta_load_cmd_buf()
1059 	cmd->cmd.cmd_load_ta.cmd_buf_len = context->mem_context.shared_mem_size;  in psp_prep_ta_load_cmd_buf()
1085 	cmd->cmd.cmd_invoke_cmd.buf.total_size = context->mem_context.shared_mem_size;  in psp_prep_ta_invoke_indirect_cmd_buf()
1086 	cmd->cmd.cmd_invoke_cmd.buf.buf_desc[0].buf_size = context->mem_context.shared_mem_size;  in psp_prep_ta_invoke_indirect_cmd_buf()
1088 				     lower_32_bits(context->mem_context.shared_mc_addr);  in psp_prep_ta_invoke_indirect_cmd_buf()
1090 				     upper_32_bits(context->mem_context.shared_mc_addr);  in psp_prep_ta_invoke_indirect_cmd_buf()
1205 	psp->xgmi_context.context.mem_context.shared_mem_size = PSP_XGMI_SHARED_MEM_SIZE;  in psp_xgmi_initialize()
1208 	if (!psp->xgmi_context.context.mem_context.shared_buf) {  in psp_xgmi_initialize()
1209 		ret = psp_ta_init_shared_buf(psp, &psp->xgmi_context.context.mem_context);  in psp_xgmi_initialize()
1223 	xgmi_cmd = (struct ta_xgmi_shared_memory *)(psp->xgmi_context.context.mem_context.shared_buf);  in psp_xgmi_initialize()
1238 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_hive_id()
1258 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_node_id()
1341 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_topology_info()
1421 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_set_topology_info()
1443 		(struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_ta_check_status()
1473 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_invoke()
1517 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_enable_features()
1614 	psp->ras_context.context.mem_context.shared_mem_size = PSP_RAS_SHARED_MEM_SIZE;  in psp_ras_initialize()
1618 		ret = psp_ta_init_shared_buf(psp, &psp->ras_context.context.mem_context);  in psp_ras_initialize()
1623 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_initialize()
1653 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_trigger_error()
1694 	psp->hdcp_context.context.mem_context.shared_mem_size = PSP_HDCP_SHARED_MEM_SIZE;  in psp_hdcp_initialize()
1698 		ret = psp_ta_init_shared_buf(psp, &psp->hdcp_context.context.mem_context);  in psp_hdcp_initialize()
1761 	psp->dtm_context.context.mem_context.shared_mem_size = PSP_DTM_SHARED_MEM_SIZE;  in psp_dtm_initialize()
1765 		ret = psp_ta_init_shared_buf(psp, &psp->dtm_context.context.mem_context);  in psp_dtm_initialize()
1829 	psp->rap_context.context.mem_context.shared_mem_size = PSP_RAP_SHARED_MEM_SIZE;  in psp_rap_initialize()
1833 		ret = psp_ta_init_shared_buf(psp, &psp->rap_context.context.mem_context);  in psp_rap_initialize()
1849 		psp_ta_free_shared_buf(&psp->rap_context.context.mem_context);  in psp_rap_initialize()
1889 		  psp->rap_context.context.mem_context.shared_buf;  in psp_rap_invoke()
1927 	psp->securedisplay_context.context.mem_context.shared_mem_size =  in psp_securedisplay_initialize()
1933 					     &psp->securedisplay_context.context.mem_context);  in psp_securedisplay_initialize()
1952 		psp_ta_free_shared_buf(&psp->securedisplay_context.context.mem_context);  in psp_securedisplay_initialize()