Unknown builtin op: spconv::get_indice_pairs

🐛 the bug

While trying to load a scripted model. I get this error.

model = torch::jit::load(settings.model_path);

Error Trackeback:


 loading model !!!!
unknown file: Failure
C++ exception with description "
Unknown builtin op: spconv::get_indice_pairs.
Could not find any similar ops to spconv::get_indice_pairs. This op may not exist or may not be currently supported in TorchScript.
:
  File "/home/ptripathi/centernet-env/lib/python3.6/site-packages/spconv/ops.py", line 79
        out_shape = spatial_shape
    if grid is None:
        res = torch.ops.spconv.get_indice_pairs(indices, batch_size, out_shape,
              ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE
                                                spatial_shape, ksize, stride,
                                                padding, dilation, out_padding,
Serialized   File "code/__torch__/spconv/ops.py", line 37
    out_shape = spatial_shape
  if torch.__is__(grid, None):
    res = ops.spconv.get_indice_pairs(indices, batch_size, out_shape, spatial_shape, ksize, stride, padding, dilation, out_padding, int(subm), int(transpose), int(use_hash))
          ~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE
    _7 = res
  else:
'get_indice_pairs' is being compiled since it was called from 'SubMConv3d.forward'
Serialized   File "code/__torch__/spconv/conv.py", line 26
  def forward(self: __torch__.spconv.conv.SubMConv3d,
    input: __torch__.spconv.spconvtensor.SparseConvTensor) -> __torch__.spconv.spconvtensor.SparseConvTensor:
    _0 = __torch__.spconv.ops.get_indice_pairs
    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE
    _1 = __torch__.spconv.functional.SubMConvFunction
    input0 = unchecked_cast(__torch__.spconv.spconvtensor.SparseConvTensor, input)
" thrown in the test body.

Versions

Collecting environment information... PyTorch version: 1.10.0+cu111 Is debug build: False CUDA used to build PyTorch: 11.1 ROCM used to build PyTorch: N/A

OS: Ubuntu 18.04.6 LTS (x86_64) GCC version: (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0 Clang version: Could not collect CMake version: version 3.13.3 Libc version: glibc-2.25

Python version: 3.6.9 (default, Jan 26 2021, 15:33:00) [GCC 8.4.0] (64-bit runtime) Python platform: Linux-5.4.0-91-generic-x86_64-with-Ubuntu-18.04-bionic Is CUDA available: True CUDA runtime version: 11.1.105 GPU models and configuration: GPU 0: Quadro P2000 Nvidia driver version: 470.82.01 cuDNN version: Probably one of the following: /usr/lib/x86_64-linux-gnu/libcudnn.so.8.0.5 /usr/lib/x86_64-linux-gnu/libcudnn_adv_infer.so.8.0.5 /usr/lib/x86_64-linux-gnu/libcudnn_adv_train.so.8.0.5 /usr/lib/x86_64-linux-gnu/libcudnn_cnn_infer.so.8.0.5 /usr/lib/x86_64-linux-gnu/libcudnn_cnn_train.so.8.0.5 /usr/lib/x86_64-linux-gnu/libcudnn_ops_infer.so.8.0.5 /usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8.0.5 HIP runtime version: N/A MIOpen runtime version: N/A

Versions of relevant libraries: [pip3] numpy==1.19.5 [pip3] torch==1.10.0+cu111 [pip3] torchaudio==0.10.0+cu111 [pip3] torchvision==0.11.0+cu111 [conda] Could not collect

pytorch / pytorch

Unknown builtin op: spconv::get_indice_pairs #70058

🐛 the bug

Versions