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

[inductor] Remove unused local variable. (#120227)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/120227
Approved by: https://github.com/Skylion007
This commit is contained in:
Taras Tsugrii
2024-04-13 03:19:10 +00:00
committed by PyTorch MergeBot
parent 2a2e1d8e4f
commit 71b8363f40
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/_inductor/comm_analysis.py
View File

@ -61,7 +61,6 @@ def get_collective_type(node: ir.IRNode) -> NCCL_COLL:
def get_collective_input_size_bytes(node: ir.IRNode) -> int:
sz_bytes = 0
for inp in node.inputs: # type: ignore[attr-defined]
shape = inp.layout.size
numel = sympy_product(inp.layout.size)
if isinstance(numel, sympy.Integer):
# For ease of testing
@ -239,7 +238,6 @@ def estimate_nccl_collective_runtime(node: ir.IRNode) -> float:
# =============== latency computation ===============
intraHw = NCCL_HW.NVLINK
hw = intraHw if nNodes == 1 else NCCL_HW.NET
if coll == NCCL_COLL.ALL_REDUCE:
if nNodes > 1: