tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Wing Lian
2024-11-12 23:20:47 -05:00
committed by GitHub
parent ad435a3b09
commit 810ebc2c0e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/environment.py
View File

@@ -15,7 +15,7 @@ def check_cuda_p2p_ib_support():
device_names, device_count = get_gpu_info() device_names, device_count = get_gpu_info()
if 1 < device_count < 8: if 1 < device_count < 8:
if any( if any(
device_name in unsupported_device unsupported_device in device_name
for device_name in device_names for device_name in device_names
for unsupported_device in unsupported_devices for unsupported_device in unsupported_devices
): ):