"...git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "d9335d2a1e3b919a627e55bd96ca050de926cecc"
Commit 84b9c0ac authored by 藍+85CD's avatar 藍+85CD
Browse files

Import intel_extension_for_pytorch as ipex

parent 1ced2bdd
...@@ -87,7 +87,7 @@ except: ...@@ -87,7 +87,7 @@ except:
pass pass
try: try:
import intel_extension_for_pytorch import intel_extension_for_pytorch as ipex
if torch.xpu.is_available(): if torch.xpu.is_available():
vram_state = XPU vram_state = XPU
except: except:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment