From 853fd5c96503b240a0f0ff86bdb229029fcc181a Mon Sep 17 00:00:00 2001 From: Shivam Gupta Date: Fri, 25 Nov 2022 21:51:56 +0530 Subject: [PATCH] Fix RuntimeError while running examples/eager_mode.py (#1647) --- python/torch_mlir_e2e_test/eager_backends/refbackend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/torch_mlir_e2e_test/eager_backends/refbackend.py b/python/torch_mlir_e2e_test/eager_backends/refbackend.py index 85f4b8c03..3449ab655 100644 --- a/python/torch_mlir_e2e_test/eager_backends/refbackend.py +++ b/python/torch_mlir_e2e_test/eager_backends/refbackend.py @@ -88,4 +88,4 @@ class EagerModeRefBackend(TorchMLIREagerBackend): return torch.from_numpy(e).clone() def transfer_from_torch_to_device(self, tensor: torch.Tensor) -> np.ndarray: - return tensor.numpy() + return tensor.detach().numpy()