del musa_utils from musa_accelerator

This commit is contained in:
KakaruHayate 2025-10-11 22:36:58 +08:00
parent 50db2e9199
commit 1f91faf51e

View File

@ -17,7 +17,6 @@ from lightning_fabric.utilities.types import _DEVICE
from pytorch_lightning.accelerators.accelerator import Accelerator from pytorch_lightning.accelerators.accelerator import Accelerator
from pytorch_lightning.utilities.exceptions import MisconfigurationException from pytorch_lightning.utilities.exceptions import MisconfigurationException
import musa_utils
# from icecream import ic # from icecream import ic
import logging import logging
@ -55,7 +54,7 @@ class MUSAAccelerator(Accelerator):
@staticmethod @staticmethod
def parse_devices(devices: Any) -> Any: def parse_devices(devices: Any) -> Any:
return musa_utils.get_device() return torch.device("musa") if _musa_available else None
@staticmethod @staticmethod
@override @override