This reverts commit 0284801fbd.
This commit is contained in:
parent
0284801fbd
commit
4c487313f2
65 changed files with 65 additions and 11 deletions
|
|
@ -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
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -15,6 +15,7 @@
|
||||||
#
|
#
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
import intel_extension_for_pytorch as ipex
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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,\
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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):
|
||||||
"""
|
"""
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue