Repository: pytorch
Issue: test_max_pool1d reliably OOMs after https://github.com/pytorch/pytorch/pull/111216/
Labels: module: memory usage,triaged,oncall: pt2

### 🐛 Describe the bug

tracker issue

sample job https://github.com/pytorch/pytorch/actions/runs/6512785700/job/17691482527

```
test/nn/test_pooling.py::TestPoolingNNDeviceTypeCPU::test_pooling_max_nhwc_cpu_float32, test/nn/test_pooling.py::TestPoolingNNDeviceTypeCPU::test_pooling_max_nhwc_cpu_float64, test/nn/test_pooling.py::TestPoolingNNDeviceTypeCPU::test_pooling_shape_cpu, test/nn/test_pooling.py::TestPoolingNNDeviceTypeCPU::test_pooling_zero_stride_cpu
stepcurrent: skipping 69 already run items.

nn/test_pooling.py::TestPoolingNNDeviceTypeCPU::test_max_pool1d_cpu_float32 LLVM ERROR: out of memory
Fatal Python error: Aborted

Current thread 0x00007fee4c1f5280 (most recent call first):
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/nn/functional.py", line 710 in _max_pool1d
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_jit_internal.py", line 499 in fn
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/utils.py", line 1462 in run_node
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/utils.py", line 1384 in <lambda>
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/utils.py", line 952 in wrap_fake_exception
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/utils.py", line 1383 in get_fake_value
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/variables/builder.py", line 1393 in wrap_fx_proxy_cls
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/variables/builder.py", line 1306 in wrap_fx_proxy
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/variables/torch.py", line 654 in call_function
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 577 in call_function
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 1190 in CALL_FUNCTION_KW
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 405 in wrapper
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 705 in step
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 742 in run
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 2353 in inline_call_
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 2229 in inline_call
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 613 in inline_user_function_return
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/variables/functions.py", line 90 in call_function
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/variables/functions.py", line 261 in call_function
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/variables/functions.py", line 307 in call_function
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 577 in call_function
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 1178 in CALL_FUNCTION_EX
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 405 in wrapper
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 705 in step
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 742 in run
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 2353 in inline_call_
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 2229 in inline_call
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 613 in inline_user_function_return
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/variables/functions.py", line 90 in call_function
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/variables/functions.py", line 261 in call_function
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/variables/nn_module.py", line 727 in call_function
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 577 in call_function
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 1138 in CALL_FUNCTION
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 405 in wrapper
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 705 in step
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 742 in run
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/symbolic_convert.py", line 2102 in run
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/convert_frame.py", line 485 in transform
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/bytecode_transformation.py", line 1028 in transform_code_object
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/convert_frame.py", line 515 in compile_inner
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/utils.py", line 221 in time_wrapper
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/convert_frame.py", line 598 in _compile
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/convert_frame.py", line 390 in _convert_frame_assert
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/convert_frame.py", line 143 in _fn
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/convert_frame.py", line 671 in _convert_frame
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/eval_frame.py", line 549 in catch_errors
  File "nn/test_pooling.py", line 840 in <resume in test_max_pool1d>
  File "nn/test_pooling.py", line 834 in <resume in test_max_pool1d>
  File "nn/test_pooling.py", line 833 in <resume in test_max_pool1d>
  File "nn/test_pooling.py", line 832 in <resume in test_max_pool1d>
  File "nn/test_pooling.py", line 831 in test_max_pool1d
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/testing/_internal/common_device_type.py", line 1084 in only_fn
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/testing/_internal/common_device_type.py", line 415 in instantiated_test
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/testing/_internal/common_utils.py", line 2453 in wrapper
  File "/opt/conda/envs/py_3.8/lib/python3.8/unittest/case.py", line 633 in _callTestMethod
  File "/opt/conda/envs/py_3.8/lib/python3.8/unittest/case.py", line 676 in run
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/external_utils.py", line 17 in inner
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/_dynamo/eval_frame.py", line 401 in _fn
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/testing/_internal/common_utils.py", line 2526 in _run_with_retry
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/testing/_internal/common_utils.py", line 2597 in run
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/testing/_internal/common_device_type.py", line 476 in run
  File "/opt/conda/envs/py_3.8/lib/python3.8/unittest/case.py", line 736 in __call__
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/_pytest/unittest.py", line 333 in runtest
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/_pytest/runner.py", line 169 in pytest_runtest_call
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pluggy/_callers.py", line 77 in _multicall
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pluggy/_manager.py", line 115 in _hookexec
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pluggy/_hooks.py", line 493 in __call__
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/_pytest/runner.py", line 262 in <lambda>
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/_pytest/runner.py", line 341 in from_call
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/_pytest/runner.py", line 261 in call_runtest_hook
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/_pytest/runner.py", line 222 in call_and_report
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/_pytest/runner.py", line 133 in runtestprotocol
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pytest_rerunfailures.py", line 608 in pytest_runtest_protocol
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pluggy/_callers.py", line 77 in _multicall
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pluggy/_manager.py", line 115 in _hookexec
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pluggy/_hooks.py", line 493 in __call__
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/_pytest/main.py", line 348 in pytest_runtestloop
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pluggy/_callers.py", line 77 in _multicall
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pluggy/_manager.py", line 115 in _hookexec
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pluggy/_hooks.py", line 493 in __call__
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/_pytest/main.py", line 323 in _main
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/_pytest/main.py", line 269 in wrap_session
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/_pytest/main.py", line 316 in pytest_cmdline_main
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pluggy/_callers.py", line 77 in _multicall
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pluggy/_manager.py", line 115 in _hookexec
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/pluggy/_hooks.py", line 493 in __call__
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/_pytest/config/__init__.py", line 166 in main
  File "/opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/testing/_internal/common_utils.py", line 965 in run_tests
  File "nn/test_pooling.py", line 1545 in <module>

FINISHED PRINTING LOG FILE of nn/test_pooling 1/1 (/var/lib/jenkins/workspace/test/test-reports/nn.test_pooling_1.1_z1q3pb3i_toprint.log)
```

### Versions

main

cc @msaroufim @wconstab @bdhirsh @anijain2305