tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-02-24 04:20:26 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: updated is_ddp to use more robust checks

Browse Source 
Refactor is_ddp function to check whether ddp is being used or not
This commit is contained in:
Ananth 2025-10-17 15:05:28 +05:30 committed by GitHub
parent d6d86cbf4c
commit 1892172251
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nanochat/common.py
View File

@ -76,8 +76,8 @@ def print_banner():
print0(banner)
def is_ddp():
# TODO is there a proper way
return int(os.environ.get('RANK', -1)) != -1
# check if DDP is available and initialized
return dist.is_available() and dist.is_initialized()
def get_dist_info():
if is_ddp():