1462 				      &psp->rap_context.rap_shared_bo,  in psp_rap_init_shared_buf()
1463 				      &psp->rap_context.rap_shared_mc_addr,  in psp_rap_init_shared_buf()
1464 				      &psp->rap_context.rap_shared_buf);  in psp_rap_init_shared_buf()
1484 				 psp->rap_context.rap_shared_mc_addr,  in psp_rap_load()
1490 		psp->rap_context.rap_initialized = true;  in psp_rap_load()
1491 		psp->rap_context.session_id = cmd->resp.session_id;  in psp_rap_load()
1492 		mutex_init(&psp->rap_context.mutex);  in psp_rap_load()
1509 	psp_prep_ta_unload_cmd_buf(cmd, psp->rap_context.session_id);  in psp_rap_unload()
1534 	if (!psp->rap_context.rap_initialized) {  in psp_rap_initialize()
1548 		amdgpu_bo_free_kernel(&psp->rap_context.rap_shared_bo,  in psp_rap_initialize()
1549 			      &psp->rap_context.rap_shared_mc_addr,  in psp_rap_initialize()
1550 			      &psp->rap_context.rap_shared_buf);  in psp_rap_initialize()
1552 		psp->rap_context.rap_initialized = false;  in psp_rap_initialize()
1565 	if (!psp->rap_context.rap_initialized)  in psp_rap_terminate()
1570 	psp->rap_context.rap_initialized = false;  in psp_rap_terminate()
1573 	amdgpu_bo_free_kernel(&psp->rap_context.rap_shared_bo,  in psp_rap_terminate()
1574 			      &psp->rap_context.rap_shared_mc_addr,  in psp_rap_terminate()
1575 			      &psp->rap_context.rap_shared_buf);  in psp_rap_terminate()
1585 	if (!psp->rap_context.rap_initialized)  in psp_rap_invoke()
1592 	mutex_lock(&psp->rap_context.mutex);  in psp_rap_invoke()
1595 		  psp->rap_context.rap_shared_buf;  in psp_rap_invoke()
1601 	ret = psp_ta_invoke(psp, rap_cmd->cmd_id, psp->rap_context.session_id);  in psp_rap_invoke()
1603 		mutex_unlock(&psp->rap_context.mutex);  in psp_rap_invoke()
1607 	mutex_unlock(&psp->rap_context.mutex);  in psp_rap_invoke()