diff --git a/src/pyronn_torch/codegen.py b/src/pyronn_torch/codegen.py index 314975fa8a36f8254d05b9c856c1eaae64fd8787..0baa4223634c030246b9ef4c74d454d0bdb53d89 100644 --- a/src/pyronn_torch/codegen.py +++ b/src/pyronn_torch/codegen.py @@ -231,8 +231,8 @@ def compile_shared_object(output_folder=None, source_files=None): torch_extension = load(module_name, [generated_file] + cuda_sources, with_cuda=True, - extra_cflags=['--std=c++14'], - extra_cuda_cflags=['-std=c++14', '-arch=sm_35'], + extra_cflags=[], + extra_cuda_cflags=['-arch=sm_35'], build_directory=join(object_cache, module_name), extra_include_paths=[]) shared_object_file = join(object_cache, module_name, module_name + '.so')