Unknown builtin op: sru_cuda::sru_bi_forward_simple

[ctlaltdefeat]

When using a bidirectional SRU, regular usage seems to be fine, and compilation to torchscript proceeds without error, but upon trying to infer with the compiled torchscript I get:

Unknown builtin op: sru_cuda::sru_bi_forward_simple.

Using pytorch 1.10, sru 2.6.0, cuda 11.3

[jinpoon]

Can you try setting CUDA_VISIBLE_DEVICES="" when you torchscript your model?

[taolei87]

Hi @ctlaltdefeat ,

Could you also provide some details:

• are you trying to use the torchscript model in Python or C++?
• are you trying to use the torchscript model with GPU or with CPU?