From d0cf1696dbc865956843e453d07bd94cb37fcae3 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 2 May 2024 22:52:11 +0300 Subject: [PATCH] IPEX getCurrentRawStream support --- 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 e5aba693c..fbd3d88e4 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._getCurrentStream + #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