From 228018e09de594e55b9549cdf0f377e36549df1e Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 2 May 2024 22:52:48 +0300 Subject: [PATCH] Cleanup --- modules/intel/ipex/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/intel/ipex/__init__.py b/modules/intel/ipex/__init__.py index fbd3d88e4..4dd5b15e9 100644 --- a/modules/intel/ipex/__init__.py +++ b/modules/intel/ipex/__init__.py @@ -147,7 +147,7 @@ def ipex_init(): # pylint: disable=too-many-statements torch.cuda.amp.GradScaler = ipex.cpu.autocast._grad_scaler.GradScaler # C - #torch._C._cuda_getCurrentRawStream = ipex._C._getCurrentRawStream + torch._C._cuda_getCurrentRawStream = ipex._C._getCurrentRawStream ipex._C._DeviceProperties.multi_processor_count = ipex._C._DeviceProperties.gpu_subslice_count ipex._C._DeviceProperties.major = 2024 ipex._C._DeviceProperties.minor = 0