454 	struct amdtee *amdtee;  in amdtee_driver_init()  local
467 	amdtee = kzalloc(sizeof(*amdtee), GFP_KERNEL);  in amdtee_driver_init()
468 	if (!amdtee) {  in amdtee_driver_init()
480 	teedev = tee_device_alloc(&amdtee_desc, NULL, pool, amdtee);  in amdtee_driver_init()
485 	amdtee->teedev = teedev;  in amdtee_driver_init()
487 	rc = tee_device_register(amdtee->teedev);  in amdtee_driver_init()
491 	amdtee->pool = pool;  in amdtee_driver_init()
493 	drv_data->amdtee = amdtee;  in amdtee_driver_init()
499 	tee_device_unregister(amdtee->teedev);  in amdtee_driver_init()
505 	kfree(amdtee);  in amdtee_driver_init()
518 	struct amdtee *amdtee;  in amdtee_driver_exit()  local
520 	if (!drv_data || !drv_data->amdtee)  in amdtee_driver_exit()
523 	amdtee = drv_data->amdtee;  in amdtee_driver_exit()
525 	tee_device_unregister(amdtee->teedev);  in amdtee_driver_exit()
526 	tee_shm_pool_free(amdtee->pool);  in amdtee_driver_exit()