From: Oded Gabbay Date: Mon, 14 Dec 2020 10:52:06 +0000 (+0200) Subject: habanalabs: register to pci shutdown callback X-Git-Tag: v5.11-rc3~7^2~5^2~2 X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=fcaebc7354188b0d708c79df4390fbabd4d9799d;p=thirdparty%2Flinux.git habanalabs: register to pci shutdown callback We need to make sure our device is idle when rebooting a virtual machine. This is done in the driver level. The firmware will later handle FLR but we want to be extra safe and stop the devices until the FLR is handled. Signed-off-by: Oded Gabbay --- diff --git a/drivers/misc/habanalabs/common/habanalabs_drv.c b/drivers/misc/habanalabs/common/habanalabs_drv.c index 6bbb6bca68600..032d114f01ea5 100644 --- a/drivers/misc/habanalabs/common/habanalabs_drv.c +++ b/drivers/misc/habanalabs/common/habanalabs_drv.c @@ -544,6 +544,7 @@ static struct pci_driver hl_pci_driver = { .id_table = ids, .probe = hl_pci_probe, .remove = hl_pci_remove, + .shutdown = hl_pci_remove, .driver.pm = &hl_pm_ops, .err_handler = &hl_pci_err_handler, };