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

Remove unused imports in test

parent 2101c570
No related branches found
No related tags found
No related merge requests found
...@@ -9,20 +9,13 @@ ...@@ -9,20 +9,13 @@
""" """
import os import os
import subprocess
from os.path import join
from sysconfig import get_paths
import pytest import pytest
import sympy import sympy
import pystencils import pystencils
from pystencils.cpu.cpujit import get_compiler_config
from pystencils.include import get_pystencils_include_path
from pystencils_autodiff import create_backward_assignments from pystencils_autodiff import create_backward_assignments
from pystencils_autodiff._file_io import write_cached_content, write_file
from pystencils_autodiff.backends.astnodes import TensorflowModule from pystencils_autodiff.backends.astnodes import TensorflowModule
from pystencils_autodiff.tensorflow_jit import _compile_env
def test_detect_cpu_vs_cpu(): def test_detect_cpu_vs_cpu():
...@@ -45,11 +38,6 @@ def test_detect_cpu_vs_cpu(): ...@@ -45,11 +38,6 @@ def test_detect_cpu_vs_cpu():
def test_native_tensorflow_compilation_cpu(): def test_native_tensorflow_compilation_cpu():
tf = pytest.importorskip('tensorflow') tf = pytest.importorskip('tensorflow')
extra_flags = ['-I' + get_paths()['include'], '-I' + get_pystencils_include_path()]
compile_flags = tf.sysconfig.get_compile_flags()
link_flags = tf.sysconfig.get_link_flags()
module_name = "Ololol" module_name = "Ololol"
target = 'cpu' target = 'cpu'
...@@ -86,11 +74,6 @@ def test_native_tensorflow_compilation_cpu(): ...@@ -86,11 +74,6 @@ def test_native_tensorflow_compilation_cpu():
def test_native_tensorflow_compilation_gpu(): def test_native_tensorflow_compilation_gpu():
tf = pytest.importorskip('tensorflow') tf = pytest.importorskip('tensorflow')
extra_flags = ['-I' + get_paths()['include'], '-I' + get_pystencils_include_path()]
compile_flags = tf.sysconfig.get_compile_flags()
link_flags = tf.sysconfig.get_link_flags()
module_name = "Ololol" module_name = "Ololol"
target = 'gpu' target = 'gpu'
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment