diff --git a/src/pyronn_torch/codegen.py b/src/pyronn_torch/codegen.py index 84ad2ebbcf9e51951bd84b2ee428f55e5f843209..63251e222c5abd2c1f89d9ddc547854e91d2f3a5 100644 --- a/src/pyronn_torch/codegen.py +++ b/src/pyronn_torch/codegen.py @@ -204,7 +204,6 @@ def generate_shared_object(output_folder=None, dst = join(output_folder, basename(s).replace('.cu.cc', '.cu')) copyfile(s, dst) # Torch only accepts *.cu as CUDA - module = framework_module_class(module_name, FUNCTIONS.values()) if show_code: @@ -228,7 +227,6 @@ def generate_shared_object(output_folder=None, return extension - def compile_shared_object(output_folder=None, source_files=None): """ Same as generate_shared_object without pystencils dependency