Type torch::jit::Module converted to torch._C.ScriptModule

jit
1

Hi,
I’m trying to develop some torch.jit modules with c++，and then use them in python with pybind.
I found that the orginal method of torch._C.ScriptModule._create_method_from_trace
was created with arg types torch::jit::Module in c++. But the corresponding method arg type was torch._c.ScriptModule in python.
How dose this conversion happens?
I try to define a method in c++ with arg type torch::jit::Module. When exported to python, it’s arg type still torch::jit::Module.
What’s missed when building the program?

2

Build with these cxx flags can fix:

 cmake_cxx_flags: List[str] = []
    for name in ["COMPILER_TYPE", "STDLIB", "BUILD_ABI"]:
        val = getattr(torch._C, f"_PYBIND11_{name}")
        if val is not None:
            cmake_cxx_flags += [f'-DPYBIND11_{name}=\\"{val}\\"']
        print("val: ", val)
        print("cmake_cxx_flags: ", cmake_cxx_flags)