diff --git a/codegen/generate_wrappers.py b/codegen/generate_wrappers.py index e149093965a452e9ae15eb01bdc930ede6338e43..9d284361990784309a0f2fe6ebddef509606a5ed 100644 --- a/codegen/generate_wrappers.py +++ b/codegen/generate_wrappers.py @@ -179,6 +179,9 @@ def main(): if hasattr(v, '__doc__'): print(v.__doc__) + shared_object_file = module.compiled_file.replace('.cpp', '.so') + copyfile(shared_object_file, join(args.output_folder, 'pyronn_torch.so')) + if __name__ == '__main__': main()