Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
cross entropy _input -> input rename
Signed-off-by: Pawel Gadzinski <[email protected]>
  • Loading branch information
pggPL committed Nov 4, 2025
commit 15c6741d26f53be134cfd457e276e5445965113b
3 changes: 1 addition & 2 deletions transformer_engine/pytorch/cross_entropy.py
Original file line number Diff line number Diff line change
Expand Up @@ -138,10 +138,9 @@ def parallel_cross_entropy(
# Handle backward compatibility with _input parameter
if _input is not None:
warnings.warn(
"The '_input' parameter is deprecated and will be removed in a future version. "
"The '_input' parameter is deprecated. "
"Please use 'input' instead.",
FutureWarning,
stacklevel=2,
)
input = _input

Expand Down
Loading