reasoning_about_shapes.py failing with

[svekars]

Add Link

reasoning_about_shapes.py

Describe the bug

faling with:

Unexpected failing examples:
2025-04-15T19:13:02.0149943Z /var/lib/workspace/recipes_source/recipes/reasoning_about_shapes.py failed leaving traceback:
2025-04-15T19:13:02.0150452Z Traceback (most recent call last):
2025-04-15T19:13:02.0150942Z   File "/var/lib/workspace/recipes_source/recipes/reasoning_about_shapes.py", line 88, in <module>
2025-04-15T19:13:02.0151433Z     out = net(inp)
2025-04-15T19:13:02.0151919Z   File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
2025-04-15T19:13:02.0152533Z     return self._call_impl(*args, **kwargs)
2025-04-15T19:13:02.0153367Z   File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 1845, in _call_impl
2025-04-15T19:13:02.0153874Z     return inner()
2025-04-15T19:13:02.0154284Z   File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 1793, in inner
2025-04-15T19:13:02.0154783Z     result = forward_call(*args, **kwargs)
2025-04-15T19:13:02.0156238Z   File "/var/lib/workspace/recipes_source/recipes/reasoning_about_shapes.py", line 62, in forward
2025-04-15T19:13:02.0156759Z     x = self.pool(F.relu(self.conv1(x)))
2025-04-15T19:13:02.0157233Z   File "/usr/local/lib/python3.10/dist-packages/torch/nn/functional.py", line 1704, in relu
2025-04-15T19:13:02.0157696Z     result = torch.relu(input)
2025-04-15T19:13:02.0158184Z   File "/usr/local/lib/python3.10/dist-packages/torch/_refs/nn/functional/__init__.py", line 147, in _fn
2025-04-15T19:13:02.0158723Z     return fn(a, *args, inplace=False, **kwargs)
2025-04-15T19:13:02.0159246Z   File "/usr/local/lib/python3.10/dist-packages/torch/_prims_common/wrappers.py", line 291, in _fn
2025-04-15T19:13:02.0159741Z     result = fn(*args, **kwargs)
2025-04-15T19:13:02.0160216Z   File "/usr/local/lib/python3.10/dist-packages/torch/_prims_common/wrappers.py", line 143, in _fn
2025-04-15T19:13:02.0160716Z     result = fn(**bound.arguments)
2025-04-15T19:13:02.0161404Z   File "/usr/local/lib/python3.10/dist-packages/torch/_refs/nn/functional/__init__.py", line 265, in relu
2025-04-15T19:13:02.0161944Z     return torch.where(torch.le(a, 0), 0, a)
2025-04-15T19:13:02.0162438Z   File "/usr/local/lib/python3.10/dist-packages/torch/_prims_common/wrappers.py", line 291, in _fn
2025-04-15T19:13:02.0162931Z     result = fn(*args, **kwargs)
2025-04-15T19:13:02.0163393Z   File "/usr/local/lib/python3.10/dist-packages/torch/_prims_common/wrappers.py", line 143, in _fn
2025-04-15T19:13:02.0163891Z     result = fn(**bound.arguments)
2025-04-15T19:13:02.0173241Z   File "/usr/local/lib/python3.10/dist-packages/torch/_refs/__init__.py", line 1980, in where
2025-04-15T19:13:02.0173753Z     pred, a, b = _maybe_broadcast(pred, a, b)
2025-04-15T19:13:02.0174285Z   File "/usr/local/lib/python3.10/dist-packages/torch/_refs/__init__.py", line 459, in _maybe_broadcast
2025-04-15T19:13:02.0175009Z     return tuple(__maybe_broadcast(x, common_shape) for x in args)
2025-04-15T19:13:02.0175586Z   File "/usr/local/lib/python3.10/dist-packages/torch/_refs/__init__.py", line 459, in <genexpr>
2025-04-15T19:13:02.0176153Z     return tuple(__maybe_broadcast(x, common_shape) for x in args)
2025-04-15T19:13:02.0176754Z   File "/usr/local/lib/python3.10/dist-packages/torch/_refs/__init__.py", line 451, in __maybe_broadcast
2025-04-15T19:13:02.0177273Z     return x.expand(common_shape)
2025-04-15T19:13:02.0177572Z ValueError: expected a tensor object
2025-04-15T19:13:02.0177773Z 

Diabled by #3325

Describe your environment

• CI
• PyTorch 2.6

cc: @mikaylagawarecki