457 	struct amdtee *amdtee;  in amdtee_driver_init()  local
470 	amdtee = kzalloc(sizeof(*amdtee), GFP_KERNEL);  in amdtee_driver_init()
471 	if (!amdtee) {  in amdtee_driver_init()
483 	teedev = tee_device_alloc(&amdtee_desc, NULL, pool, amdtee);  in amdtee_driver_init()
488 	amdtee->teedev = teedev;  in amdtee_driver_init()
490 	rc = tee_device_register(amdtee->teedev);  in amdtee_driver_init()
494 	amdtee->pool = pool;  in amdtee_driver_init()
496 	drv_data->amdtee = amdtee;  in amdtee_driver_init()
502 	tee_device_unregister(amdtee->teedev);  in amdtee_driver_init()
508 	kfree(amdtee);  in amdtee_driver_init()
521 	struct amdtee *amdtee;  in amdtee_driver_exit()  local
523 	if (!drv_data || !drv_data->amdtee)  in amdtee_driver_exit()
526 	amdtee = drv_data->amdtee;  in amdtee_driver_exit()
528 	tee_device_unregister(amdtee->teedev);  in amdtee_driver_exit()
529 	tee_shm_pool_free(amdtee->pool);  in amdtee_driver_exit()