mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-07-14 03:07:24 +00:00
Fix python lint formatting in internode multirank test
Agent-Logs-Url: https://github.com/microsoft/mscclpp/sessions/f5220581-e26c-49d8-98fa-e1b8ab011898 Co-authored-by: seagater <7475084+seagater@users.noreply.github.com>
This commit is contained in:
committed by
GitHub
parent
757c5ec831
commit
04ea24da8d
@@ -96,6 +96,7 @@ def main():
|
||||
|
||||
# Small settings for functional check
|
||||
import os as _os
|
||||
|
||||
num_tokens = int(_os.environ.get("MSCCLPP_EP_BENCH_TOKENS", "128"))
|
||||
hidden = int(_os.environ.get("MSCCLPP_EP_BENCH_HIDDEN", "1024"))
|
||||
num_topk = int(_os.environ.get("MSCCLPP_EP_BENCH_TOPK", str(min(4, num_ranks))))
|
||||
|
||||
Reference in New Issue
Block a user