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

axolotl-ai-cloud · Nov 13, 2024 · 810ebc2 · 810ebc2
1 parent ad435a3
commit 810ebc2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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
             ):