From 810ebc2c0e6a723069a2648796795361c5559470 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 12 Nov 2024 23:20:47 -0500 Subject: [PATCH] invert the string in string check for p2p device check (#2044) --- src/axolotl/utils/environment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/environment.py b/src/axolotl/utils/environment.py index c447b1ee25..cf2e5d23d2 100644 --- 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 ):