gistfile1.txt

import torch
import torch.onnx

class MyCastModule(torch.nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self, x):
        x = x.int()
        return torch.add(x, x)

mcm = MyCastModule()
mcm.eval()

x = torch.zeros(1, 2, 3).long()

torch.onnx._export(mcm, x, 'test.onnx', verbose=True)

################

$ python casttest.py

Traceback (most recent call last):
  File "casttest.py", line 17, in <module>
    torch.onnx._export(mcm, x, 'test.onnx', verbose=True)
  File "/Users/jamesreed/onnx-fairseq/pytorch/torch/onnx/__init__.py", line 132, in _export
    trace, torch_out = torch.jit.get_trace_graph(model, args)
  File "/Users/jamesreed/onnx-fairseq/pytorch/torch/jit/__init__.py", line 252, in get_trace_graph
    return LegacyTracedModule(f, nderivs=nderivs)(*args, **kwargs)
  File "/Users/jamesreed/onnx-fairseq/pytorch/torch/nn/modules/module.py", line 363, in __call__
    result = self.forward(*input, **kwargs)
  File "/Users/jamesreed/onnx-fairseq/pytorch/torch/jit/__init__.py", line 288, in forward
    torch._C._tracer_exit(out_vars)
RuntimeError: /Users/jamesreed/onnx-fairseq/pytorch/torch/csrc/jit/tracer.h:117: getTracingState: Assertion `state` failed.