Revert "[LLM] IPEX auto importer turn on by default for XPU (#9730)" (#9759)

This reverts commit 0284801fbd.
This commit is contained in:
Qiyuan Gong 2023-12-22 16:38:24 +08:00 committed by GitHub
parent 0284801fbd
commit 4c487313f2
65 changed files with 65 additions and 11 deletions

View file

@ -25,6 +25,7 @@ world_size = int(os.getenv("WORLD_SIZE", "1"))
from bigdl.llm import optimize_model from bigdl.llm import optimize_model
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -17,6 +17,7 @@
import torch import torch
import time import time
import argparse import argparse
import intel_extension_for_pytorch as ipex
from bigdl.llm.transformers import AutoModelForCausalLM from bigdl.llm.transformers import AutoModelForCausalLM
from transformers import AutoTokenizer from transformers import AutoTokenizer

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -17,6 +17,7 @@
import torch import torch
import time import time
import argparse import argparse
import intel_extension_for_pytorch as ipex
from bigdl.llm.transformers import AutoModelForCausalLM from bigdl.llm.transformers import AutoModelForCausalLM
from transformers import LlamaTokenizer, GPTQConfig from transformers import LlamaTokenizer, GPTQConfig

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse
import numpy as np import numpy as np

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse
import numpy as np import numpy as np

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse
import numpy as np import numpy as np

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse
import numpy as np import numpy as np

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -14,6 +14,7 @@
# limitations under the License. # limitations under the License.
# #
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse
import numpy as np import numpy as np

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse
import numpy as np import numpy as np

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse
import numpy as np import numpy as np

View file

@ -22,6 +22,7 @@ from transformers import AutoTokenizer
from transformers.generation import GenerationConfig from transformers.generation import GenerationConfig
from bigdl.llm.transformers import AutoModelForCausalLM from bigdl.llm.transformers import AutoModelForCausalLM
import intel_extension_for_pytorch as ipex
torch.manual_seed(1234) torch.manual_seed(1234)

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -17,6 +17,7 @@
import os import os
import torch import torch
import time import time
import intel_extension_for_pytorch as ipex
import argparse import argparse
import numpy as np import numpy as np
import inquirer import inquirer

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import argparse import argparse
from bigdl.llm.transformers import AutoModelForCausalLM from bigdl.llm.transformers import AutoModelForCausalLM
from transformers import AutoTokenizer, TextGenerationPipeline from transformers import AutoTokenizer, TextGenerationPipeline

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse
import numpy as np import numpy as np

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse
import numpy as np import numpy as np

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -14,6 +14,7 @@
# limitations under the License. # limitations under the License.
# #
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -23,6 +23,7 @@ import time
import warnings import warnings
from typing import Dict, List, Optional, Union from typing import Dict, List, Optional, Union
import intel_extension_for_pytorch as ipex
import torch import torch
import torch.nn as nn import torch.nn as nn
from accelerate import init_empty_weights from accelerate import init_empty_weights

View file

@ -34,6 +34,7 @@
import argparse import argparse
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
from transformers import AutoModelForCausalLM from transformers import AutoModelForCausalLM

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -22,6 +22,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
from transformers.generation import GenerationConfig from transformers.generation import GenerationConfig
from bigdl.llm import optimize_model from bigdl.llm import optimize_model
import intel_extension_for_pytorch as ipex
torch.manual_seed(1234) torch.manual_seed(1234)

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse

View file

@ -15,6 +15,7 @@
# #
import torch import torch
import intel_extension_for_pytorch as ipex
import time import time
import argparse import argparse
from bigdl.llm import optimize_model from bigdl.llm import optimize_model

View file

@ -46,6 +46,7 @@ from peft import (
) )
from utils.prompter import Prompter from utils.prompter import Prompter
import intel_extension_for_pytorch as ipex
from bigdl.llm.transformers import AutoModelForCausalLM from bigdl.llm.transformers import AutoModelForCausalLM
# import them from bigdl.llm.transformers.qlora to get a BigDL-LLM compatible Peft model # import them from bigdl.llm.transformers.qlora to get a BigDL-LLM compatible Peft model
from bigdl.llm.transformers.qlora import get_peft_model, prepare_model_for_kbit_training,\ from bigdl.llm.transformers.qlora import get_peft_model, prepare_model_for_kbit_training,\

View file

@ -19,6 +19,7 @@ import os
import transformers import transformers
from transformers import LlamaTokenizer from transformers import LlamaTokenizer
import intel_extension_for_pytorch as ipex
from peft import LoraConfig from peft import LoraConfig
from bigdl.llm.transformers.qlora import get_peft_model, prepare_model_for_kbit_training from bigdl.llm.transformers.qlora import get_peft_model, prepare_model_for_kbit_training
from bigdl.llm.transformers import AutoModelForCausalLM from bigdl.llm.transformers import AutoModelForCausalLM

View file

@ -24,7 +24,7 @@ from .optimize import optimize_model
import os import os
# Default is false, set to true to auto importing Intel Extension for PyTorch. # Default is false, set to true to auto importing Intel Extension for PyTorch.
BIGDL_IMPORT_IPEX = os.getenv("BIGDL_IMPORT_IPEX", 'True').lower() in ('true', '1', 't') BIGDL_IMPORT_IPEX = os.getenv("BIGDL_IMPORT_IPEX", 'False').lower() in ('true', '1', 't')
if BIGDL_IMPORT_IPEX: if BIGDL_IMPORT_IPEX:
# Import Intel Extension for PyTorch as ipex if XPU version is installed # Import Intel Extension for PyTorch as ipex if XPU version is installed
from .utils.ipex_importer import ipex_importer from .utils.ipex_importer import ipex_importer

View file

@ -35,19 +35,10 @@ class IPEXImporter:
""" """
# Check if xpu version installed # Check if xpu version installed
try: try:
# Check if bigdl-core-xe is installed
distribution('bigdl-core-xe') distribution('bigdl-core-xe')
return True return True
except PackageNotFoundError: except PackageNotFoundError:
# bigdl-core-xe not found return False
# Check if bigdl-core-xe-21 is installed
try:
distribution('bigdl-core-xe-21')
return True
except PackageNotFoundError:
# bigdl-core-xe not found
return False
return False
def import_ipex(self): def import_ipex(self):
""" """