72 static void set_ta_context_funcs(struct psp_context *psp,  in set_ta_context_funcs()  argument
78 		*pcontext = &psp->ras_context.context;  in set_ta_context_funcs()
79 		psp->ta_funcs = &ras_ta_funcs;  in set_ta_context_funcs()
153 	struct psp_context   *psp     = &adev->psp;  in ta_if_load_debugfs_write()  local
179 	set_ta_context_funcs(psp, ta_type, &context);  in ta_if_load_debugfs_write()
181 	if (!psp->ta_funcs || !psp->ta_funcs->fn_ta_terminate) {  in ta_if_load_debugfs_write()
192 		ret = psp_ta_init_shared_buf(psp, &context->mem_context);  in ta_if_load_debugfs_write()
199 	ret = psp_fn_ta_terminate(psp);  in ta_if_load_debugfs_write()
215 	if (!psp->ta_funcs->fn_ta_initialize) {  in ta_if_load_debugfs_write()
221 	ret = psp_fn_ta_initialize(psp);  in ta_if_load_debugfs_write()
251 	struct psp_context   *psp     = &adev->psp;  in ta_if_unload_debugfs_write()  local
267 	set_ta_context_funcs(psp, ta_type, &context);  in ta_if_unload_debugfs_write()
270 	if (!psp->ta_funcs || !psp->ta_funcs->fn_ta_terminate) {  in ta_if_unload_debugfs_write()
275 	ret = psp_fn_ta_terminate(psp);  in ta_if_unload_debugfs_write()
300 	struct psp_context   *psp     = &adev->psp;  in ta_if_invoke_debugfs_write()  local
330 	set_ta_context_funcs(psp, ta_type, &context);  in ta_if_invoke_debugfs_write()
338 	if (!psp->ta_funcs || !psp->ta_funcs->fn_ta_invoke) {  in ta_if_invoke_debugfs_write()
346 	mutex_lock(&psp->ras_context.mutex);  in ta_if_invoke_debugfs_write()
351 	ret = psp_fn_ta_invoke(psp, cmd_id);  in ta_if_invoke_debugfs_write()
365 	mutex_unlock(&psp->ras_context.mutex);  in ta_if_invoke_debugfs_write()