pytorch
e7d795dc - [Inductor] aten.{avg_pool2d/max_pool2d_with_indices} arguments can be 1 element tuple (#96727)

Commit
1 year ago
[Inductor] aten.{avg_pool2d/max_pool2d_with_indices} arguments can be 1 element tuple (#96727) Fixes failure from 14k github models: ```pytest ./generated/test_ProGamerGov_neural_dream.py -k test_000``` Error: ``` ...... File "/scratch/ybliang/work/repos/pytorch/torch/_inductor/graph.py", line 357, in call_function raise LoweringException(e, target, args, kwargs).with_traceback( File "/scratch/ybliang/work/repos/pytorch/torch/_inductor/graph.py", line 354, in call_function out = lowerings[target](*args, **kwargs) File "/scratch/ybliang/work/repos/pytorch/torch/_inductor/lowering.py", line 228, in wrapped out = decomp_fn(*args, **kwargs) File "/scratch/ybliang/work/repos/pytorch/torch/_inductor/lowering.py", line 3124, in avg_pool2d assert len(padding) == 2 torch._dynamo.exc.BackendCompilerFailed: backend='inductor' raised: LoweringException: AssertionError: target: aten.avg_pool2d.default args[0]: TensorBox(StorageBox( InputBuffer(name='arg0_1', layout=FixedLayout('cuda', torch.float32, size=[4, 4, 64, 64], stride=[16384, 4096, 64, 1])) )) args[1]: [7, 7] args[2]: [1, 1] args[3]: [0] args[4]: False args[5]: False ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/96727 Approved by: https://github.com/jansel
Author
Committer
Parents
Loading