issue/365 - fixing an interface mismatch

1d1e0649 · thatPepe · GitHub · adbda4c4 · 1d1e0649
Unverified Commit 1d1e0649 authored Aug 14, 2025 by thatPepe Committed by GitHub Aug 14, 2025
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
@@ -608,6 +608,6 @@ def get_sync_func(device):
    if device == InfiniDeviceEnum.CPU or device == InfiniDeviceEnum.CAMBRICON:
        sync = None
    else:
-        sync = getattr(torch, infiniDeviceEnum_str_map[device]).synchronize
+        sync = getattr(torch, torch_device_map[device]).synchronize

    return sync