Dear friend，
I realy appreciate your help. However,I still have a question about how wrap custom methods.
temp.py
import torch
import base_cpp
from torch.fx import symbolic_trace
import torch.fx

def pybase_cpp(x):
    return base_cpp.forward(x, torch.randn(1), torch.randn(1))  # base_cpp.forward no __code__

torch.fx.wrap('base_cpp.forward')
pybase_cpp(torch.randn(1))
traced = symbolic_trace(pybase_cpp)

error
KeyError: 'base_cpp.forward'

So. Is there any good way to wrap custom methods. Thanks very much.