Skip to content
Snippets Groups Projects
Commit d1c464ac authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Return JIT compiled extension

parent c868f90a
Branches
Tags
No related merge requests found
......@@ -178,14 +178,12 @@ def generate_shared_object(output_folder=None, source_files=None, show_code=Fals
extension = module.compile(extra_source_files=cuda_sources, extra_cuda_flags=['-arch=sm_35'], with_cuda=True)
for v in extension.__dict__.values():
if hasattr(v, '__doc__'):
print(v.__doc__)
shared_object_file = module.compiled_file.replace('.cpp', '.so')
copyfile(shared_object_file, join(output_folder, 'pyronn_torch.so'))
copyfile(module.compiled_file, join(output_folder, 'pyronn_torch.cpp'))
return extension
def main():
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment