Merge pull request #217 from wooway777/issue/216

issue/216 - Fixed Enum Camparison in util.py

Merge pull request #217 from wooway777/issue/216
issue/216 - Fixed Enum Camparison in util.py
e580d751 · PanZezhong1725 · GitHub · 0c803397 · b2b4526b · e580d751
Unverified Commit e580d751 authored May 16, 2025 by PanZezhong1725 Committed by GitHub May 16, 2025
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test/infiniop/libinfiniop/utils.py test/infiniop/libinfiniop/utils.py +1 -1

No files found.
--- a/test/infiniop/libinfiniop/utils.py
+++ b/test/infiniop/libinfiniop/utils.py
@@ -478,7 +478,7 @@ def get_sync_func(device):
    import torch
    device_str = infiniDeviceEnum_str_map[device]
    
-    if device_str == "cpu":
+    if device == InfiniDeviceEnum.CPU:
        sync = None
    else:
        sync = getattr(torch, device_str).synchronize