invert the string in string check for p2p device check (#2044)

2024-11-12 23:20:47 -05:00
parent ad435a3b09
commit 810ebc2c0e
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/utils/environment.py
+++ b/src/axolotl/utils/environment.py
@@ -15,7 +15,7 @@ def check_cuda_p2p_ib_support():
        device_names, device_count = get_gpu_info()
        if 1 < device_count < 8:
            if any(
-                device_name in unsupported_device
+                unsupported_device in device_name
                for device_name in device_names
                for unsupported_device in unsupported_devices
            ):