Traceback (most recent call last): File "/builds/software-machine-learning-infra-frameworks-workspaces-robhar02/pytorch/test/test_jit.py", line 15805, in test_conv_error fn(torch.ones(2, 2), torch.ones(4, 4)) File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 946, in prof_func_call return prof_callable(func_call, *args, **kwargs) File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 938, in prof_callable return callable(*args, kwargs) RuntimeError: The following operation failed in the TorchScript interpreter. Traceback of TorchScript (most recent call last): File "/builds/software-machine-learning-infra-frameworks-workspaces-robhar02/pytorch/test/test_jit.py", line 15802, in fn @torch.jit.script def fn(x, y): return F.conv2d(x, y) ~~~~~~~~ <--- HERE RuntimeError: Expected 3D (unbatched) or 4D (batched) input to conv2d, but got input of size: [2, 2] Expected 3D (unbatched) or 4D (batched) input to conv2d, but got input of size: [2, 2] Exception raised from batchify at /builds/software-machine-learning-infra-frameworks-workspaces-robhar02/pytorch/aten/src/ATen/native/Convolution.cpp:821 (most recent call first): frame #0: c10::Error::Error(c10::SourceLocation, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >) + 0xc8 (0xffff9eb36778 in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libc10.so) frame #1: c10::detail::torchCheckFail(char const, char const, unsigned int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > const&) + 0xc4 (0xffff9ead9b6c in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libc10.so) frame #2: <unknown function> + 0x13c8430 (0xffff9fbc8430 in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so) frame #3: at::native::conv2d_symint(at::Tensor const&, at::Tensor const&, std::optionalat::Tensor const&, c10::ArrayRefc10::SymInt, c10::ArrayRefc10::SymInt, c10::ArrayRefc10::SymInt, c10::SymInt) + 0xa4 (0xffff9fbca6a4 in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so) frame #4: <unknown function> + 0x2c9ebb4 (0xffffa149ebb4 in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so) frame #5: <unknown function> + 0x63b7e8c (0xffffa4bb7e8c in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so) frame #6: <unknown function> + 0x5f9154c (0xffffa479154c in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so) frame #7: torch::jit::InterpreterState::run(std::vector<c10::IValue, std::allocatorc10::IValue >&) + 0x48 (0xffffa47950e8 in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so) frame #8: <unknown function> + 0x5f74364 (0xffffa4774364 in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libtorch_cpu.so) frame #9: <unknown function> + 0xdef04c (0xffffa9fbf04c in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libtorch_python.so) frame #10: <unknown function> + 0xeebc0c (0xffffaa0bbc0c in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libtorch_python.so) frame #11: <unknown function> + 0xeec284 (0xffffaa0bc284 in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libtorch_python.so) frame #12: <unknown function> + 0x5c7104 (0xffffa9797104 in /opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/lib/libtorch_python.so) frame #13: <unknown function> + 0x22416c (0xaaaabf1c416c in /opt/conda/envs/py_3.10/bin/python) frame #14: _PyObject_Call + 0x68 (0xaaaabf0149f0 in /opt/conda/envs/py_3.10/bin/python) frame #15: _PyObject_Call + 0x68 (0xaaaabf0149f0 in /opt/conda/envs/py_3.10/bin/python) frame #16: _PyEval_EvalFrameDefault + 0x3ac8 (0xaaaabf0b6754 in /opt/conda/envs/py_3.10/bin/python) frame #17: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #18: _PyEval_EvalFrameDefault + 0x3ac8 (0xaaaabf0b6754 in /opt/conda/envs/py_3.10/bin/python) frame #19: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #20: _PyObject_FastCallDictTstate + 0x74 (0xaaaabf014eb4 in /opt/conda/envs/py_3.10/bin/python) frame #21: _PyObject_Call_Prepend + 0x140 (0xaaaabf015240 in /opt/conda/envs/py_3.10/bin/python) frame #22: <unknown function> + 0xc78fc (0xaaaabf0678fc in /opt/conda/envs/py_3.10/bin/python) frame #23: _PyObject_MakeTpCall + 0xa0 (0xaaaabf014cc0 in /opt/conda/envs/py_3.10/bin/python) frame #24: _PyEval_EvalFrameDefault + 0x5534 (0xaaaabf0b81c0 in /opt/conda/envs/py_3.10/bin/python) frame #25: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #26: <unknown function> + 0x20aea8 (0xaaaabf1aaea8 in /opt/conda/envs/py_3.10/bin/python) frame #27: _PyEval_EvalFrameDefault + 0x3ac8 (0xaaaabf0b6754 in /opt/conda/envs/py_3.10/bin/python) frame #28: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #29: <unknown function> + 0x20ae28 (0xaaaabf1aae28 in /opt/conda/envs/py_3.10/bin/python) frame #30: _PyEval_EvalFrameDefault + 0x54e0 (0xaaaabf0b816c in /opt/conda/envs/py_3.10/bin/python) frame #31: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #32: _PyEval_EvalFrameDefault + 0x5968 (0xaaaabf0b85f4 in /opt/conda/envs/py_3.10/bin/python) frame #33: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #34: <unknown function> + 0x20ae28 (0xaaaabf1aae28 in /opt/conda/envs/py_3.10/bin/python) frame #35: _PyEval_EvalFrameDefault + 0x52d0 (0xaaaabf0b7f5c in /opt/conda/envs/py_3.10/bin/python) frame #36: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #37: <unknown function> + 0x20ae28 (0xaaaabf1aae28 in /opt/conda/envs/py_3.10/bin/python) frame #38: _PyEval_EvalFrameDefault + 0x52d0 (0xaaaabf0b7f5c in /opt/conda/envs/py_3.10/bin/python) frame #39: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #40: <unknown function> + 0x20ae28 (0xaaaabf1aae28 in /opt/conda/envs/py_3.10/bin/python) frame #41: PyVectorcall_Call + 0xe8 (0xaaaabf0148b4 in /opt/conda/envs/py_3.10/bin/python) frame #42: _PyEval_EvalFrameDefault + 0x3ac8 (0xaaaabf0b6754 in /opt/conda/envs/py_3.10/bin/python) frame #43: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #44: _PyObject_FastCallDictTstate + 0x144 (0xaaaabf014f84 in /opt/conda/envs/py_3.10/bin/python) frame #45: _PyObject_Call_Prepend + 0x140 (0xaaaabf015240 in /opt/conda/envs/py_3.10/bin/python) frame #46: <unknown function> + 0xc78fc (0xaaaabf0678fc in /opt/conda/envs/py_3.10/bin/python) frame #47: _PyObject_MakeTpCall + 0xa0 (0xaaaabf014cc0 in /opt/conda/envs/py_3.10/bin/python) frame #48: _PyEval_EvalFrameDefault + 0x6864 (0xaaaabf0b94f0 in /opt/conda/envs/py_3.10/bin/python) frame #49: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #50: _PyEval_EvalFrameDefault + 0x5968 (0xaaaabf0b85f4 in /opt/conda/envs/py_3.10/bin/python) frame #51: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #52: _PyEval_EvalFrameDefault + 0x3ac8 (0xaaaabf0b6754 in /opt/conda/envs/py_3.10/bin/python) frame #53: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #54: _PyEval_EvalFrameDefault + 0x6bdc (0xaaaabf0b9868 in /opt/conda/envs/py_3.10/bin/python) frame #55: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #56: <unknown function> + 0x20ae28 (0xaaaabf1aae28 in /opt/conda/envs/py_3.10/bin/python) frame #57: _PyEval_EvalFrameDefault + 0x6bdc (0xaaaabf0b9868 in /opt/conda/envs/py_3.10/bin/python) frame #58: <unknown function> + 0x11bb40 (0xaaaabf0bbb40 in /opt/conda/envs/py_3.10/bin/python) frame #59: _PyObject_FastCallDictTstate + 0x144 (0xaaaabf014f84 in /opt/conda/envs/py_3.10/bin/python) frame #60: _PyObject_Call_Prepend + 0x140 (0xaaaabf015240 in /opt/conda/envs/py_3.10/bin/python) frame #61: <unknown function> + 0xc78fc (0xaaaabf0678fc in /opt/conda/envs/py_3.10/bin/python) frame #62: _PyObject_Call + 0x68 (0xaaaabf0149f0 in /opt/conda/envs/py_3.10/bin/python)

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "/builds/software-machine-learning-infra-frameworks-workspaces-robhar02/pytorch/test/test_jit.py", line 15807, in test_conv_error self.assertFalse('frame' in str(e)) File "/opt/conda/envs/py_3.10/lib/python3.10/unittest/case.py", line 681, in assertFalse raise self.failureException(msg) AssertionError: True is not false

To execute this test, run the following from the base repo dir: python test/test_jit.py TestScript.test_conv_error

Fix Plan

The fix involves modifying the input to the conv2d function to have the correct number of dimensions.

• Check the input dimensions: The input to conv2d should be 3D (unbatched) or 4D (batched).
• Modify the input: Use torch.unsqueeze to add a dimension to the input if necessary.

Example code:

import torch
import torch.nn.functional as F

# Define the input
x = torch.ones(2, 2)
y = torch.ones(4, 4)

# Add a dimension to the input if necessary
x = torch.unsqueeze(x, 0)  # batch dimension
y = torch.unsqueeze(y, 0)  # batch dimension

# Define the function with the correct input
@torch.jit.script
def fn(x, y):
    return F.conv2d(x, y)

# Call the function
result = fn(x, y)

Verification

To verify the fix, run the test again:

python test/test_jit.py TestScript.test_conv_error

The test should pass without any errors.

Extra Tips

• Always check the input dimensions before calling a function like conv2d.
• Use torch.unsqueeze to add dimensions to the input if necessary.
• Make sure to test the function with different input sizes to ensure it works correctly in all cases.