issue/216 - Fixed Enum Camparison in util.py

3c696932 · wooway777 · 125afeb5 · 3c696932
Commit 3c696932 authored May 13, 2025 by wooway777
Hide 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
@@ -477,7 +477,7 @@ def get_test_devices(args):
 def get_sync_func(device):
    import torch
    
-    if device == "cpu":
+    if device == InfiniDeviceEnum.CPU:
        sync = None
    else:
        sync = getattr(torch, infiniDeviceEnum_str_map[device]).synchronize