optimize(infer): move ipex into rvc

configs/config.py

@@ -6,16 +6,7 @@ import shutil
 from multiprocessing import cpu_count
 
 import torch
-
+# TODO: move device selection into rvc
-try:
-    import intel_extension_for_pytorch as ipex  # pylint: disable=import-error, unused-import
-
-    if torch.xpu.is_available():
-        from infer.modules.ipex import ipex_init
-
-        ipex_init()
-except Exception:  # pylint: disable=broad-exception-caught
-    pass
 import logging
 
 logger = logging.getLogger(__name__)

infer/modules/train/train.py

@@ -24,8 +24,7 @@ try:
     import intel_extension_for_pytorch as ipex  # pylint: disable=import-error, unused-import
 
     if torch.xpu.is_available():
-        from infer.modules.ipex import ipex_init
+        from rvc.ipex import ipex_init, gradscaler_init
-        from infer.modules.ipex.gradscaler import gradscaler_init
         from torch.xpu.amp import autocast
 
         GradScaler = gradscaler_init()

rvc/__init__.py

@@ -0,0 +1,4 @@
+from . import ipex
+import sys
+del sys.modules["rvc.ipex"]
+

rvc/ipex/__init__.py

@@ -0,0 +1,8 @@
+try:
+    import torch
+    if torch.xpu.is_available():
+        from .init import ipex_init
+        ipex_init()
+        from .gradscaler import gradscaler_init
+except Exception:  # pylint: disable=broad-exception-caught
+    pass

rvc/ipex/gradscaler.py

@@ -1,4 +1,5 @@
 from collections import defaultdict
+
 import torch
 import intel_extension_for_pytorch as ipex  # pylint: disable=import-error, unused-import
 import intel_extension_for_pytorch._C as core  # pylint: disable=import-error, unused-import

rvc/ipex/hijacks.py

@@ -1,5 +1,6 @@
 import contextlib
 import importlib
+
 import torch
 import intel_extension_for_pytorch as ipex  # pylint: disable=import-error, unused-import
 

rvc/ipex/init.py

@@ -1,8 +1,10 @@
 import os
 import sys
 import contextlib
+
 import torch
 import intel_extension_for_pytorch as ipex  # pylint: disable=import-error, unused-import
+
 from .hijacks import ipex_hijacks
 from .attention import attention_init