Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Sep 18, 2024
1 parent 8199c45 commit cd186e7
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion transformer_engine/pytorch/module/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,7 +153,8 @@ def initialize_ub(
s.close()
else:
ifname_warning = (
f"'{ifname}' is not a valid network interface! `te.initialize_ub()` will attempt to "
f"'{ifname}' is not a valid network interface! `te.initialize_ub()` will"
" attempt to "
+ "detect ranks on the same node by matching 'socket.gethostname()', which is "
+ "known to fail on virtual clusters like Kubernetes. If Userbuffers "
+ "initialization fails, please set the 'NVTE_UB_SOCKET_IFNAME' variable in "
Expand Down

0 comments on commit cd186e7

Please sign in to comment.