🐛 [Bug] isBool() INTERNAL ASSERT FAILED while compiling Sequence Generator

[Csinclair0]

Bug Description

I am receving this error when compling a torchscript module. The module is a light wrapper around a SequenceGenerator from fairseq.

RuntimeError: isBool() INTERNAL ASSERT FAILED at "bazel-out/k8-opt/bin/external/libtorch_pre_cxx11_abi/_virtual_includes/ATen/ATen/core/ivalue.h":625, please report a bug to PyTorch. 

To Reproduce

I am currently executing sequence generation in torch, using the compile spec.

trt_ts_module = torch_tensorrt.ts.compile(generator,
  inputs=[    
          torch_tensorrt.Input(min_shape=(1,4), opt_shape = (1,10), max_shape = (1,20) , dtype = torch.int32),  
      ],   
      debug= True, 
      enabled_precisions={torch.half},    
      truncate_long_and_double=True,
      require_full_compilation=False, 
      torch_executed_modules= ["fairseq.sequence_generator.SequenceGenerator"], 
      min_block_size = 1
  )

Expected behavior

no compile errors

Environment

building docker image from Dockerfile in repo.

[narendasan]

Can you share the logs from the failure?

[Csinclair0]

Here are the full_logs.

[narendasan]

Looking at the logs, it seems like this error pops up during graph splitting. @peri044 Can you take a look or delegate this to someone?

[peri044]

RuntimeError: isBool() INTERNAL ASSERT FAILED at "bazel-out/k8-opt/bin/external/libtorch_pre_cxx11_abi/_virtual_includes/ATen/ATen/core/ivalue.h":625, please report a bug to PyTorch. 

Looks like we some converter args failure. @bowang007 Can you take a look at this ?

[bowang007]

sure. Let me check what's going on

[bowang007]

Hey @Csinclair0 This bug was fixed by this PR: #1691
This PR was merged into our master branch last week, please take a took and try if it works.
Thanks!