295 	psp_ta_free_shared_buf(&psp->xgmi_context.context.mem_context);  in psp_free_shared_bufs()
298 	psp_ta_free_shared_buf(&psp->ras_context.context.mem_context);  in psp_free_shared_bufs()
301 	psp_ta_free_shared_buf(&psp->hdcp_context.context.mem_context);  in psp_free_shared_bufs()
304 	psp_ta_free_shared_buf(&psp->dtm_context.context.mem_context);  in psp_free_shared_bufs()
307 	psp_ta_free_shared_buf(&psp->rap_context.context.mem_context);  in psp_free_shared_bufs()
310 	psp_ta_free_shared_buf(&psp->securedisplay_context.context.mem_context);  in psp_free_shared_bufs()
1124 	psp->asd_context.mem_context.shared_mc_addr  = 0;  in psp_asd_initialize()
1125 	psp->asd_context.mem_context.shared_mem_size = PSP_ASD_SHARED_MEM_SIZE;  in psp_asd_initialize()
1214 		lower_32_bits(context->mem_context.shared_mc_addr);  in psp_prep_ta_load_cmd_buf()
1216 		upper_32_bits(context->mem_context.shared_mc_addr);  in psp_prep_ta_load_cmd_buf()
1217 	cmd->cmd.cmd_load_ta.cmd_buf_len = context->mem_context.shared_mem_size;  in psp_prep_ta_load_cmd_buf()
1327 	psp->xgmi_context.context.mem_context.shared_mem_size = PSP_XGMI_SHARED_MEM_SIZE;  in psp_xgmi_initialize()
1330 	if (!psp->xgmi_context.context.mem_context.shared_buf) {  in psp_xgmi_initialize()
1331 		ret = psp_ta_init_shared_buf(psp, &psp->xgmi_context.context.mem_context);  in psp_xgmi_initialize()
1345 	xgmi_cmd = (struct ta_xgmi_shared_memory *)(psp->xgmi_context.context.mem_context.shared_buf);  in psp_xgmi_initialize()
1362 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_hive_id()
1382 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_node_id()
1471 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_topology_info()
1588 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_set_topology_info()
1610 		(struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_ta_check_status()
1646 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_send_cmd()
1700 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_invoke()
1845 	psp->ras_context.context.mem_context.shared_mem_size = PSP_RAS_SHARED_MEM_SIZE;  in psp_ras_initialize()
1848 	if (!psp->ras_context.context.mem_context.shared_buf) {  in psp_ras_initialize()
1849 		ret = psp_ta_init_shared_buf(psp, &psp->ras_context.context.mem_context);  in psp_ras_initialize()
1854 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_initialize()
1974 	psp->hdcp_context.context.mem_context.shared_mem_size = PSP_HDCP_SHARED_MEM_SIZE;  in psp_hdcp_initialize()
1977 	if (!psp->hdcp_context.context.mem_context.shared_buf) {  in psp_hdcp_initialize()
1978 		ret = psp_ta_init_shared_buf(psp, &psp->hdcp_context.context.mem_context);  in psp_hdcp_initialize()
2048 	psp->dtm_context.context.mem_context.shared_mem_size = PSP_DTM_SHARED_MEM_SIZE;  in psp_dtm_initialize()
2051 	if (!psp->dtm_context.context.mem_context.shared_buf) {  in psp_dtm_initialize()
2052 		ret = psp_ta_init_shared_buf(psp, &psp->dtm_context.context.mem_context);  in psp_dtm_initialize()
2119 	psp->rap_context.context.mem_context.shared_mem_size = PSP_RAP_SHARED_MEM_SIZE;  in psp_rap_initialize()
2122 	if (!psp->rap_context.context.mem_context.shared_buf) {  in psp_rap_initialize()
2123 		ret = psp_ta_init_shared_buf(psp, &psp->rap_context.context.mem_context);  in psp_rap_initialize()
2139 		psp_ta_free_shared_buf(&psp->rap_context.context.mem_context);  in psp_rap_initialize()
2179 		  psp->rap_context.context.mem_context.shared_buf;  in psp_rap_invoke()
2221 	psp->securedisplay_context.context.mem_context.shared_mem_size =  in psp_securedisplay_initialize()
2227 					     &psp->securedisplay_context.context.mem_context);  in psp_securedisplay_initialize()
2251 		psp_ta_free_shared_buf(&psp->securedisplay_context.context.mem_context);  in psp_securedisplay_initialize()