habanalabs: prevent device PTE read/write during hard-reset
authorOded Gabbay <oded.gabbay@gmail.com>
Sat, 6 Apr 2019 12:33:38 +0000 (15:33 +0300)
committerOded Gabbay <oded.gabbay@gmail.com>
Sat, 6 Apr 2019 12:33:38 +0000 (15:33 +0300)
During hard-reset, contexts are closed as part of the tear-down process.
After a context is closed, the driver cleans up the page tables of that
context in the device's DRAM. This action is both dangerous and
unnecessary.

It is unnecessary, because the device is going through a hard-reset, which
means the device's DRAM contents are no longer valid and the device's MMU
is being reset.

It is dangerous, because if the hard-reset came as a result of a PCI
freeze, this action may cause the entire host machine to hang.

Therefore, prevent all device PTE updates when a hard-reset operation is
pending.

Signed-off-by: Oded Gabbay <oded.gabbay@gmail.com>
drivers/misc/habanalabs/goya/goya.c

index aaa05662ca9f8a6f7ed35796c90632b4407779c5..5100dfbf3acc346d9ecb9c70a3fbf779453331fd 100644 (file)
@@ -4058,6 +4058,9 @@ static u64 goya_read_pte(struct hl_device *hdev, u64 addr)
 {
        struct goya_device *goya = hdev->asic_specific;
 
+       if (hdev->hard_reset_pending)
+               return U64_MAX;
+
        return readq(hdev->pcie_bar[DDR_BAR_ID] +
                        (addr - goya->ddr_bar_cur_addr));
 }
@@ -4066,6 +4069,9 @@ static void goya_write_pte(struct hl_device *hdev, u64 addr, u64 val)
 {
        struct goya_device *goya = hdev->asic_specific;
 
+       if (hdev->hard_reset_pending)
+               return;
+
        writeq(val, hdev->pcie_bar[DDR_BAR_ID] +
                        (addr - goya->ddr_bar_cur_addr));
 }