259 	psp_ta_free_shared_buf(&psp->xgmi_context.context.mem_context);  in psp_free_shared_bufs()
262 	psp_ta_free_shared_buf(&psp->ras_context.context.mem_context);  in psp_free_shared_bufs()
265 	psp_ta_free_shared_buf(&psp->hdcp_context.context.mem_context);  in psp_free_shared_bufs()
268 	psp_ta_free_shared_buf(&psp->dtm_context.context.mem_context);  in psp_free_shared_bufs()
271 	psp_ta_free_shared_buf(&psp->rap_context.context.mem_context);  in psp_free_shared_bufs()
274 	psp_ta_free_shared_buf(&psp->securedisplay_context.context.mem_context);  in psp_free_shared_bufs()
1043 	psp->asd_context.mem_context.shared_mc_addr  = 0;  in psp_asd_initialize()
1044 	psp->asd_context.mem_context.shared_mem_size = PSP_ASD_SHARED_MEM_SIZE;  in psp_asd_initialize()
1133 		lower_32_bits(context->mem_context.shared_mc_addr);  in psp_prep_ta_load_cmd_buf()
1135 		upper_32_bits(context->mem_context.shared_mc_addr);  in psp_prep_ta_load_cmd_buf()
1136 	cmd->cmd.cmd_load_ta.cmd_buf_len = context->mem_context.shared_mem_size;  in psp_prep_ta_load_cmd_buf()
1246 	psp->xgmi_context.context.mem_context.shared_mem_size = PSP_XGMI_SHARED_MEM_SIZE;  in psp_xgmi_initialize()
1249 	if (!psp->xgmi_context.context.mem_context.shared_buf) {  in psp_xgmi_initialize()
1250 		ret = psp_ta_init_shared_buf(psp, &psp->xgmi_context.context.mem_context);  in psp_xgmi_initialize()
1264 	xgmi_cmd = (struct ta_xgmi_shared_memory *)(psp->xgmi_context.context.mem_context.shared_buf);  in psp_xgmi_initialize()
1279 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_hive_id()
1299 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_node_id()
1383 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_topology_info()
1466 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_set_topology_info()
1488 		(struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_ta_check_status()
1518 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_invoke()
1560 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_enable_features()
1656 	psp->ras_context.context.mem_context.shared_mem_size = PSP_RAS_SHARED_MEM_SIZE;  in psp_ras_initialize()
1659 	if (!psp->ras_context.context.mem_context.shared_buf) {  in psp_ras_initialize()
1660 		ret = psp_ta_init_shared_buf(psp, &psp->ras_context.context.mem_context);  in psp_ras_initialize()
1665 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_initialize()
1723 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_trigger_error()
1765 	psp->hdcp_context.context.mem_context.shared_mem_size = PSP_HDCP_SHARED_MEM_SIZE;  in psp_hdcp_initialize()
1768 	if (!psp->hdcp_context.context.mem_context.shared_buf) {  in psp_hdcp_initialize()
1769 		ret = psp_ta_init_shared_buf(psp, &psp->hdcp_context.context.mem_context);  in psp_hdcp_initialize()
1832 	psp->dtm_context.context.mem_context.shared_mem_size = PSP_DTM_SHARED_MEM_SIZE;  in psp_dtm_initialize()
1835 	if (!psp->dtm_context.context.mem_context.shared_buf) {  in psp_dtm_initialize()
1836 		ret = psp_ta_init_shared_buf(psp, &psp->dtm_context.context.mem_context);  in psp_dtm_initialize()
1900 	psp->rap_context.context.mem_context.shared_mem_size = PSP_RAP_SHARED_MEM_SIZE;  in psp_rap_initialize()
1903 	if (!psp->rap_context.context.mem_context.shared_buf) {  in psp_rap_initialize()
1904 		ret = psp_ta_init_shared_buf(psp, &psp->rap_context.context.mem_context);  in psp_rap_initialize()
1920 		psp_ta_free_shared_buf(&psp->rap_context.context.mem_context);  in psp_rap_initialize()
1960 		  psp->rap_context.context.mem_context.shared_buf;  in psp_rap_invoke()
1998 	psp->securedisplay_context.context.mem_context.shared_mem_size =  in psp_securedisplay_initialize()
2004 					     &psp->securedisplay_context.context.mem_context);  in psp_securedisplay_initialize()
2028 		psp_ta_free_shared_buf(&psp->securedisplay_context.context.mem_context);  in psp_securedisplay_initialize()