habanalabs: Add check for pci_enable_device

As the potential failure of the pci_enable_device(),
it should be better to check the return value and return
error if fails.

Fixes: 70b2f993ea ("habanalabs: create common folder")
Signed-off-by: Jiasheng Jiang <jiasheng@iscas.ac.cn>
Reviewed-by: Oded Gabbay <ogabbay@kernel.org>
Signed-off-by: Oded Gabbay <ogabbay@kernel.org>
This commit is contained in:
Jiasheng Jiang 2022-02-24 14:58:05 +08:00 committed by Oded Gabbay
parent a78b07dcae
commit 9c27896ac1

View File

@ -890,6 +890,8 @@ static ssize_t hl_set_power_state(struct file *f, const char __user *buf,
pci_set_power_state(hdev->pdev, PCI_D0); pci_set_power_state(hdev->pdev, PCI_D0);
pci_restore_state(hdev->pdev); pci_restore_state(hdev->pdev);
rc = pci_enable_device(hdev->pdev); rc = pci_enable_device(hdev->pdev);
if (rc < 0)
return rc;
} else if (value == 2) { } else if (value == 2) {
pci_save_state(hdev->pdev); pci_save_state(hdev->pdev);
pci_disable_device(hdev->pdev); pci_disable_device(hdev->pdev);