Skip to content

Commit b4e1975

Browse files
JDBetteridgeJDBetteridge
committed
lint: First pass with ruff --fix
1 parent d8696fe commit b4e1975

File tree

130 files changed

+1217
-900
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

130 files changed

+1217
-900
lines changed

‎benchmarks/user/advisor/roofline.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def roofline(name, project, scale, precision, mode, th):
130130
# y = bandwidth * x
131131
x1, x2 = 0, min(width, max_compute_bandwidth / bandwidth)
132132
y1, y2 = 0, x2*bandwidth
133-
label = '{} {:.0f} GB/s'.format(roof.name, bandwidth)
133+
label = f'{roof.name} {bandwidth:.0f} GB/s'
134134
ax.plot([x1, x2], [y1, y2], '-', label=label)
135135
memory_roofs.append(((x1, x2), (y1, y2)))
136136

@@ -140,7 +140,7 @@ def roofline(name, project, scale, precision, mode, th):
140140
bandwidth /= scale # scale down as requested by the user
141141
x1, x2 = max(bandwidth / max_memory_bandwidth, 0), width
142142
y1, y2 = bandwidth, bandwidth
143-
label = '{} {:.0f} GFLOPS'.format(roof.name, bandwidth)
143+
label = f'{roof.name} {bandwidth:.0f} GFLOPS'
144144
ax.plot([x1, x2], [y1, y2], '-', label=label)
145145
compute_roofs.append(((x1, x2), (y1, y2)))
146146

‎devito/arch/archinfo.py‎

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def get_cpu_info():
4848

4949
# Obtain textual cpu info
5050
try:
51-
with open('/proc/cpuinfo', 'r') as f:
51+
with open('/proc/cpuinfo') as f:
5252
lines = f.readlines()
5353
except FileNotFoundError:
5454
lines = []
@@ -696,9 +696,7 @@ def get_platform():
696696
elif 'intel' in brand:
697697
# Most likely a desktop i3/i5/i7
698698
return platform_registry['intel64']
699-
elif 'power8' in brand:
700-
return platform_registry['power8']
701-
elif 'power9' in brand:
699+
elif 'power8' in brand or 'power9' in brand:
702700
return platform_registry['power8']
703701
elif 'arm' in brand:
704702
return platform_registry['arm']

‎devito/arch/compiler.py‎

Lines changed: 18 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from devito.tools import (as_list, change_directory, filter_ordered,
2323
memoized_func, make_tempdir)
2424

25-
__all__ = ['sniff_mpi_distro', 'compiler_registry']
25+
__all__ = ['compiler_registry', 'sniff_mpi_distro']
2626

2727

2828
@memoized_func
@@ -50,11 +50,7 @@ def sniff_compiler_version(cc, allow_fail=False):
5050
ver = ver.strip()
5151
if ver.startswith("gcc"):
5252
compiler = "gcc"
53-
elif ver.startswith("clang"):
54-
compiler = "clang"
55-
elif ver.startswith("Apple LLVM"):
56-
compiler = "clang"
57-
elif ver.startswith("Homebrew clang"):
53+
elif ver.startswith("clang") or ver.startswith("Apple LLVM") or ver.startswith("Homebrew clang"):
5854
compiler = "clang"
5955
elif ver.startswith("Intel"):
6056
compiler = "icx"
@@ -334,22 +330,21 @@ def make(self, loc, args):
334330
logfile = path.join(self.get_jit_dir(), f"{hash_key}.log")
335331
errfile = path.join(self.get_jit_dir(), f"{hash_key}.err")
336332

337-
with change_directory(loc):
338-
with open(logfile, "w") as lf:
339-
with open(errfile, "w") as ef:
340-
341-
command = ['make'] + args
342-
lf.write("Compilation command:\n")
343-
lf.write(" ".join(command))
344-
lf.write("\n\n")
345-
try:
346-
check_call(command, stderr=ef, stdout=lf)
347-
except CalledProcessError as e:
348-
raise CompilationError(f'Command "{e.cmd}" return error status'
349-
f'{e.returncode}. '
350-
f'Unable to compile code.\n'
351-
f'Compile log in {logfile}\n'
352-
f'Compile errors in {errfile}\n')
333+
with change_directory(loc), open(logfile, "w") as lf:
334+
with open(errfile, "w") as ef:
335+
336+
command = ['make'] + args
337+
lf.write("Compilation command:\n")
338+
lf.write(" ".join(command))
339+
lf.write("\n\n")
340+
try:
341+
check_call(command, stderr=ef, stdout=lf)
342+
except CalledProcessError as e:
343+
raise CompilationError(f'Command "{e.cmd}" return error status'
344+
f'{e.returncode}. '
345+
f'Unable to compile code.\n'
346+
f'Compile log in {logfile}\n'
347+
f'Compile errors in {errfile}\n')
353348
debug(f"Make <{' '.join(args)}>")
354349

355350
def _cmdline(self, files, object=False):
@@ -387,7 +382,7 @@ def jit_compile(self, soname, code):
387382
# Warning: dropping `code` on the floor in favor to whatever is written
388383
# within `src_file`
389384
try:
390-
with open(src_file, 'r') as f:
385+
with open(src_file) as f:
391386
code = f.read()
392387
code = f'{code}/* Backdoor edit at {time.ctime()}*/ \n'
393388
# Bypass the devito JIT cache

‎devito/builtins/arithmetic.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import devito as dv
44
from devito.builtins.utils import make_retval, check_builtins_args
55

6-
__all__ = ['norm', 'sumall', 'sum', 'inner', 'mmin', 'mmax']
6+
__all__ = ['inner', 'mmax', 'mmin', 'norm', 'sum', 'sumall']
77

88

99
@dv.switchconfig(log_level='ERROR')

‎devito/builtins/initializers.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from devito.tools import as_tuple, as_list
55
from devito.builtins.utils import check_builtins_args, nbl_to_padsize, pad_outhalo
66

7-
__all__ = ['assign', 'smooth', 'gaussian_smooth', 'initialize_function']
7+
__all__ = ['assign', 'gaussian_smooth', 'initialize_function', 'smooth']
88

99

1010
@dv.switchconfig(log_level='ERROR')

‎devito/builtins/utils.py‎

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,13 @@
77
from devito.symbolics import uxreplace
88
from devito.tools import as_tuple
99

10-
__all__ = ['make_retval', 'nbl_to_padsize', 'pad_outhalo', 'abstract_args',
11-
'check_builtins_args']
10+
__all__ = [
11+
'abstract_args',
12+
'check_builtins_args',
13+
'make_retval',
14+
'nbl_to_padsize',
15+
'pad_outhalo',
16+
]
1217

1318

1419
accumulator_mapper = {

‎devito/core/arm.py‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,12 @@
22
Cpu64AdvCOperator)
33
from devito.passes.iet import OmpTarget, CXXOmpTarget
44

5-
__all__ = ['ArmAdvCOperator', 'ArmAdvOmpOperator', 'ArmAdvCXXOperator',
6-
'ArmAdvCXXOmpOperator']
5+
__all__ = [
6+
'ArmAdvCOperator',
7+
'ArmAdvCXXOmpOperator',
8+
'ArmAdvCXXOperator',
9+
'ArmAdvOmpOperator',
10+
]
711

812

913
ArmAdvOperator = Cpu64AdvOperator

‎devito/core/cpu.py‎

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,20 @@
1414
check_stability)
1515
from devito.tools import timed_pass
1616

17-
__all__ = ['Cpu64NoopCOperator', 'Cpu64NoopOmpOperator', 'Cpu64AdvCOperator',
18-
'Cpu64AdvOmpOperator', 'Cpu64FsgCOperator', 'Cpu64FsgOmpOperator',
19-
'Cpu64CustomOperator', 'Cpu64CustomCXXOperator', 'Cpu64AdvCXXOperator',
20-
'Cpu64AdvCXXOmpOperator', 'Cpu64FsgCXXOperator', 'Cpu64FsgCXXOmpOperator']
17+
__all__ = [
18+
'Cpu64AdvCOperator',
19+
'Cpu64AdvCXXOmpOperator',
20+
'Cpu64AdvCXXOperator',
21+
'Cpu64AdvOmpOperator',
22+
'Cpu64CustomCXXOperator',
23+
'Cpu64CustomOperator',
24+
'Cpu64FsgCOperator',
25+
'Cpu64FsgCXXOmpOperator',
26+
'Cpu64FsgCXXOperator',
27+
'Cpu64FsgOmpOperator',
28+
'Cpu64NoopCOperator',
29+
'Cpu64NoopOmpOperator',
30+
]
2131

2232

2333
class Cpu64OperatorMixin:

‎devito/core/gpu.py‎

Lines changed: 17 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,23 @@
1515
relax_incr_dimensions, check_stability)
1616
from devito.tools import as_tuple, timed_pass
1717

18-
__all__ = ['DeviceNoopOperator', 'DeviceAdvOperator', 'DeviceCustomOperator',
19-
'DeviceNoopOmpOperator', 'DeviceAdvOmpOperator', 'DeviceFsgOmpOperator',
20-
'DeviceCustomOmpOperator', 'DeviceNoopAccOperator', 'DeviceAdvAccOperator',
21-
'DeviceFsgAccOperator', 'DeviceCustomAccOperator', 'DeviceNoopCXXOmpOperator',
22-
'DeviceAdvCXXOmpOperator', 'DeviceFsgCXXOmpOperator',
23-
'DeviceCustomCXXOmpOperator']
18+
__all__ = [
19+
'DeviceAdvAccOperator',
20+
'DeviceAdvCXXOmpOperator',
21+
'DeviceAdvOmpOperator',
22+
'DeviceAdvOperator',
23+
'DeviceCustomAccOperator',
24+
'DeviceCustomCXXOmpOperator',
25+
'DeviceCustomOmpOperator',
26+
'DeviceCustomOperator',
27+
'DeviceFsgAccOperator',
28+
'DeviceFsgCXXOmpOperator',
29+
'DeviceFsgOmpOperator',
30+
'DeviceNoopAccOperator',
31+
'DeviceNoopCXXOmpOperator',
32+
'DeviceNoopOmpOperator',
33+
'DeviceNoopOperator',
34+
]
2435

2536

2637
class DeviceOperatorMixin:

‎devito/core/intel.py‎

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,16 @@
33
Cpu64AdvCXXOperator, Cpu64AdvCXXOmpOperator,
44
Cpu64FsgCXXOperator, Cpu64FsgCXXOmpOperator)
55

6-
__all__ = ['Intel64AdvCOperator', 'Intel64AdvOmpOperator', 'Intel64FsgCOperator',
7-
'Intel64FsgOmpOperator', 'Intel64CXXAdvCOperator', 'Intel64AdvCXXOmpOperator',
8-
'Intel64FsgCXXOperator', 'Intel64FsgCXXOmpOperator']
6+
__all__ = [
7+
'Intel64AdvCOperator',
8+
'Intel64AdvCXXOmpOperator',
9+
'Intel64AdvOmpOperator',
10+
'Intel64CXXAdvCOperator',
11+
'Intel64FsgCOperator',
12+
'Intel64FsgCXXOmpOperator',
13+
'Intel64FsgCXXOperator',
14+
'Intel64FsgOmpOperator',
15+
]
916

1017

1118
Intel64AdvCOperator = Cpu64AdvCOperator

0 commit comments

Comments
 (0)