Skip to content
Merged
Prev Previous commit
Next Next commit
update
  • Loading branch information
ShawnXuan committed Sep 4, 2024
commit 942556f157360116797afc6b80998abf5f1e7fc8
4 changes: 2 additions & 2 deletions libai/utils/distributed.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,14 +77,14 @@ def _init_distributed_env(self, cfg):
import oneflow_npu
except ImportError:
raise ImportError(
"The module 'oneflow_npu' is not installed. Please install it to use NPU devices."
"'oneflow_npu' is missing. Install it to use NPU devices."
)
elif self._device_type == "xpu":
try:
import oneflow_xpu
except ImportError:
raise ImportError(
"The module 'oneflow_xpu' is not installed. Please install it to use XPU devices."
"'oneflow_xpu' is missing. Install it to use NPU devices."
)
elif self._device_type not in ("cuda", "npu", "xpu", "cpu"):
raise NotImplementedError(f"Unsupported device {self._device_type}")
Expand Down