|
1 | | -from functools import partial |
2 | | -from hashlib import sha1 |
3 | | -from os import environ, path, makedirs |
4 | | -from packaging.version import Version |
5 | | -from subprocess import (DEVNULL, PIPE, CalledProcessError, check_output, |
6 | | - check_call, run) |
7 | 1 | import platform |
8 | | -import warnings |
9 | 2 | import time |
| 3 | +import warnings |
| 4 | +from functools import partial |
| 5 | +from hashlib import sha1 |
| 6 | +from os import environ, makedirs, path |
| 7 | +from subprocess import CalledProcessError, check_call, check_output, DEVNULL, PIPE, run |
10 | 8 |
|
11 | 9 | import numpy.ctypeslib as npct |
12 | 10 | from codepy.jit import compile_from_string |
13 | | -from codepy.toolchain import (GCCToolchain, |
14 | | - call_capture_output as _call_capture_output) |
| 11 | +from codepy.toolchain import call_capture_output as _call_capture_output |
| 12 | +from codepy.toolchain import GCCToolchain |
| 13 | +from packaging.version import Version |
15 | 14 |
|
16 | | -from devito.arch import (AMDGPUX, Cpu64, AppleArm, NvidiaDevice, POWER8, POWER9, |
17 | | - Graviton, Cortex, IntelDevice, get_nvidia_cc, NvidiaArm, |
18 | | - check_cuda_runtime, get_m1_llvm_path) |
| 15 | +from devito.arch import ( |
| 16 | + AMDGPUX, AppleArm, check_cuda_runtime, Cortex, Cpu64, get_m1_llvm_path, get_nvidia_cc, |
| 17 | + Graviton, IntelDevice, NvidiaArm, NvidiaDevice, POWER8, POWER9 |
| 18 | +) |
19 | 19 | from devito.exceptions import CompilationError |
20 | 20 | from devito.logger import debug, warning |
21 | 21 | from devito.parameters import configuration |
22 | | -from devito.tools import (as_list, change_directory, filter_ordered, |
23 | | - memoized_func, make_tempdir) |
| 22 | +from devito.tools import ( |
| 23 | + as_list, change_directory, filter_ordered, make_tempdir, memoized_func |
| 24 | +) |
24 | 25 |
|
25 | 26 | __all__ = ['sniff_mpi_distro', 'compiler_registry'] |
26 | 27 |
|
|
0 commit comments