Use generate_shared_object in main of codegen.py

b349749e · Stephan Seitz · c5a35d60 · b349749e
Commit b349749e authored 5 years ago by Stephan Seitz
--- a/src/pyronn_torch/codegen.py
+++ b/src/pyronn_torch/codegen.py
@@ -268,8 +268,8 @@ def main():
    parser.add_argument('--source-files',  default=None)
    args = parser.parse_args()

-    # generate_shared_object(args.output_folder, args.source_files, show_code=True)
-    compile_shared_object(args.output_folder, args.source_files)
+    generate_shared_object(args.output_folder, args.source_files, show_code=True)
+    # compile_shared_object(args.output_folder, args.source_files)


 if __name__ == '__main__':