Unverified Commit e580d751 authored by PanZezhong1725's avatar PanZezhong1725 Committed by GitHub
Browse files

Merge pull request #217 from wooway777/issue/216

issue/216 - Fixed Enum Camparison in util.py 
parents 0c803397 b2b4526b
...@@ -478,7 +478,7 @@ def get_sync_func(device): ...@@ -478,7 +478,7 @@ def get_sync_func(device):
import torch import torch
device_str = infiniDeviceEnum_str_map[device] device_str = infiniDeviceEnum_str_map[device]
if device_str == "cpu": if device == InfiniDeviceEnum.CPU:
sync = None sync = None
else: else:
sync = getattr(torch, device_str).synchronize sync = getattr(torch, device_str).synchronize
......
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