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

[PT2][Inductor] Change the split cat log to debug (#120823)

Browse Source 
Summary: Address the report in https://github.com/pytorch/pytorch/issues/120771.

Test Plan: see signal

Differential Revision: D54323475

Pull Request resolved: https://github.com/pytorch/pytorch/pull/120823
Approved by: https://github.com/jackiexu1992
This commit is contained in:
Menglu Yu
2024-03-01 22:34:23 +00:00
committed by PyTorch MergeBot
parent 797d4fbdf4
commit fd2ab1f613
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
torch/_inductor/fx_passes/split_cat.py
View File

@ -115,10 +115,10 @@ def normalize_split_base(
graph = match.graph
split_input, split_size, split_dim = _get_split_args(split_node)
if split_input is None or split_dim is None or split_size is None:
log.info("couldn't find split args")
log.debug("couldn't find split args")
return
if "example_value" not in split_node.meta:
log.warning("example value absent for node: %s", split_node)
log.debug("example value absent for node: %s", split_node)
return
assert isinstance(split_node.meta["example_value"], (list, tuple))
split_sections = [t.size()[split_dim] for t in split_node.meta["example_value"]]
@ -180,10 +180,10 @@ def normalize_unbind_default(match: Match, *args, **kwargs):
else:
dim = 0
if input is None:
log.info("couldn't find unbind args")
log.debug("couldn't find unbind args")
return
if "example_value" not in input.meta:
log.warning("example value absent for node: %s", input)
log.debug("example value absent for node: %s", input)
return
ndim = input.meta["example_value"].ndim
if dim < 0: # Normalize unbind dim
@ -219,12 +219,12 @@ def normalize_cat_default(match: Match, *args, **kwargs):
else:
cat_dim = 0
if tensors is None or cat_dim is None:
log.info("couldn't find cat args")
log.debug("couldn't find cat args")
return
assert isinstance(tensors, (list, tuple))
for tensor in itertools.chain([cat_node], tensors):
if "example_value" not in tensor.meta:
log.warning("example value absent for node: %s", tensor)
log.debug("example value absent for node: %s", tensor)
return
ndim = cat_node.meta["example_value"].dim()
@ -264,14 +264,14 @@ def normalize_stack_default(match: Match, *args, **kwargs):
tensors = get_arg_value(node, 0, "tensors")
dim = get_arg_value(node, 1, "dim") or 0
if tensors is None or dim is None:
log.info("couldn't find stack args")
log.debug("couldn't find stack args")
return
assert isinstance(tensors, (list, tuple))
# A bug in pytorch, some nodes miss the example_value metadata
for tensor in itertools.chain([node], tensors):
if "example_value" not in tensor.meta:
log.warning("example value absent for node: %s", tensor)
log.debug("example value absent for node: %s", tensor)
return
ndim = node.meta["example_value"].dim()