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