From 009135f9d29b1206260774374e3ad7b64472b20d Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 3 Jan 2020 13:31:28 +0100
Subject: [PATCH] Lint
---
src/pystencils_autodiff/backends/astnodes.py | 15 ++++++++-------
1 file changed, 8 insertions(+), 7 deletions(-)
diff --git a/src/pystencils_autodiff/backends/astnodes.py b/src/pystencils_autodiff/backends/astnodes.py
index 62eb5e9..51e378a 100644
--- a/src/pystencils_autodiff/backends/astnodes.py
+++ b/src/pystencils_autodiff/backends/astnodes.py
@@ -206,11 +206,12 @@ setup_pybind11(cfg)
except Exception as e:
print(e)
torch_extension = load(hash,
- [file_name],
- with_cuda=self.is_cuda,
- extra_cflags=['--std=c++14', get_compiler_config()['flags'].replace('--std=c++11', '')],
- extra_cuda_cflags=['-std=c++14', '-ccbin', get_compiler_config()['command']],
- build_directory=cache_dir,
- extra_include_paths=[get_pycuda_include_path(),
- get_pystencils_include_path()])
+ [file_name],
+ with_cuda=self.is_cuda,
+ extra_cflags=['--std=c++14',
+ get_compiler_config()['flags'].replace('--std=c++11', '')],
+ extra_cuda_cflags=['-std=c++14', '-ccbin', get_compiler_config()['command']],
+ build_directory=cache_dir,
+ extra_include_paths=[get_pycuda_include_path(),
+ get_pystencils_include_path()])
return torch_extension
--
GitLab