frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE][Ez]: Optimize unnecessary lambda with operator (#154722)

Browse Source 
Automated edits performed by FURB118. Operator is implemented in C and way faster when passed to another C method like sorted, max etc as a `key=`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/154722
Approved by: https://github.com/jansel
This commit is contained in:
Aaron Gokaslan
2025-05-30 23:47:06 +00:00
committed by PyTorch MergeBot
parent 0f3db20132
commit bbda22e648
12 changed files with 31 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
torch/distributed/pipelining/microbatch.py
View File

@ -1,6 +1,7 @@
# mypy: allow-untyped-defs
# Copyright (c) Meta Platforms, Inc. and affiliates
import logging
import operator
from typing import Any, Optional
import torch
@ -46,7 +47,7 @@ class _LossReducer(_CustomReducer):
pass
sum_reducer = _LossReducer(torch.tensor(0.0), lambda a, b: a + b)
sum_reducer = _LossReducer(torch.tensor(0.0), operator.add)
# Default chunking dimension is 0. This is used for the case where the user did
# not specify a chunking dimension.