summaryrefslogtreecommitdiff
path: root/drivers
diff options
context:
space:
mode:
authorOded Gabbay <ogabbay@kernel.org>2020-12-14 12:52:06 +0200
committerGreg Kroah-Hartman <gregkh@linuxfoundation.org>2021-01-19 18:26:14 +0100
commit93aef8e6cc08391d5c288177dcb46fe1fd88f240 (patch)
tree58059da1953986214e8c3c34491c42bdd936366b /drivers
parent79df21218d639f560ae2e9dfb89a76757ea334d8 (diff)
habanalabs: register to pci shutdown callback
[ Upstream commit fcaebc7354188b0d708c79df4390fbabd4d9799d ] 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 <ogabbay@kernel.org> Signed-off-by: Sasha Levin <sashal@kernel.org>
Diffstat (limited to 'drivers')
-rw-r--r--drivers/misc/habanalabs/habanalabs_drv.c1
1 files changed, 1 insertions, 0 deletions
diff --git a/drivers/misc/habanalabs/habanalabs_drv.c b/drivers/misc/habanalabs/habanalabs_drv.c
index 8c342fb499ca..ae50bd55f30a 100644
--- a/drivers/misc/habanalabs/habanalabs_drv.c
+++ b/drivers/misc/habanalabs/habanalabs_drv.c
@@ -443,6 +443,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,
};