From bb8368e52ea6ff5b2792b41185c4647e5ed2953a Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 14 Feb 2025 18:04:50 +0300 Subject: [PATCH] Revert torch.xpu.mem_get_info --- modules/intel/ipex/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/intel/ipex/__init__.py b/modules/intel/ipex/__init__.py index 99537fb20..a36664bb3 100644 --- a/modules/intel/ipex/__init__.py +++ b/modules/intel/ipex/__init__.py @@ -189,8 +189,8 @@ def ipex_init(): # pylint: disable=too-many-statements torch._C._XpuDeviceProperties.minor = 1 # Fix functions with ipex: - if not hasattr(torch.xpu, "mem_get_info"): - torch.xpu.mem_get_info = lambda device=None: [(torch.xpu.get_device_properties(device).total_memory - torch.xpu.memory_reserved(device)), torch.xpu.get_device_properties(device).total_memory] + # torch.xpu.mem_get_info always returns the total memory as free memory + torch.xpu.mem_get_info = lambda device=None: [(torch.xpu.get_device_properties(device).total_memory - torch.xpu.memory_reserved(device)), torch.xpu.get_device_properties(device).total_memory] torch.cuda.mem_get_info = torch.xpu.mem_get_info torch._utils._get_available_device_type = lambda: "xpu" torch.has_cuda = True