File size: 230 Bytes
c7b379a
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
try:
    import torch

    if torch.xpu.is_available():
        from .init import ipex_init

        ipex_init()
        from .gradscaler import gradscaler_init
except Exception:  # pylint: disable=broad-exception-caught
    pass