929 	return psp_ta_init_shared_buf(psp, &psp->xgmi_context.context.mem_context,  in psp_xgmi_init_shared_buf()
975 				 psp->xgmi_context.context.mem_context.shared_mc_addr,  in psp_xgmi_load()
982 		psp->xgmi_context.context.initialized = true;  in psp_xgmi_load()
983 		psp->xgmi_context.context.session_id = cmd->resp.session_id;  in psp_xgmi_load()
1008 	psp_prep_ta_unload_cmd_buf(cmd, psp->xgmi_context.context.session_id);  in psp_xgmi_unload()
1020 	return psp_ta_invoke(psp, ta_cmd_id, psp->xgmi_context.context.session_id);  in psp_xgmi_invoke()
1027 	if (!psp->xgmi_context.context.initialized)  in psp_xgmi_terminate()
1034 	psp->xgmi_context.context.initialized = false;  in psp_xgmi_terminate()
1037 	psp_ta_free_shared_buf(&psp->xgmi_context.context.mem_context);  in psp_xgmi_terminate()
1055 	if (!psp->xgmi_context.context.initialized) {  in psp_xgmi_initialize()
1068 	xgmi_cmd = (struct ta_xgmi_shared_memory *)(psp->xgmi_context.context.mem_context.shared_buf);  in psp_xgmi_initialize()
1083 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_hive_id()
1103 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_node_id()
1149 		mirror_top_info = &mirror_adev->psp.xgmi_context.top_info;  in psp_xgmi_reflect_topology_info()
1184 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_topology_info()
1244 			if (psp->xgmi_context.supports_extended_data &&  in psp_xgmi_get_topology_info()
1264 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_set_topology_info()
2746 	    psp->xgmi_context.context.initialized) {  in psp_suspend()
3142 		adev->psp.xgmi_context.supports_extended_data = false;  in psp_init_sos_base_fw()
3157 		adev->psp.xgmi_context.supports_extended_data = true;  in psp_init_sos_base_fw()