diff --git a/library/ipex/hijacks.py b/library/ipex/hijacks.py index 62d29605..4a9a3569 100644 --- a/library/ipex/hijacks.py +++ b/library/ipex/hijacks.py @@ -117,6 +117,7 @@ def linalg_solve(A, B, *args, **kwargs): # pylint: disable=invalid-name else: return original_linalg_solve(A, B, *args, **kwargs) +@property def is_cuda(self): return self.device.type == 'xpu'