Skip to content

Commit 02cfddd

Browse files
committed
minor: fix flake8
1 parent 94abe4c commit 02cfddd

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

pylops/signalprocessing/_fourierradon2d_cuda.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,11 @@
1-
from cmath import exp
21
from math import pi
32

43
import cupy as cp
54
from numba import cuda
65

76
TWO_PI_MINUS = cp.float32(-2.0 * pi)
87
TWO_PI_PLUS = cp.float32(2.0 * pi)
9-
I = cp.complex64(1j)
8+
IMG = cp.complex64(1j)
109

1110

1211
@cuda.jit
@@ -25,7 +24,7 @@ def _radon_inner_2d_kernel(x, y, f, px, h, flim0, flim1, npx, nh):
2524
# y[ih, ifr] += x[ipx, ifr] * exp(TWO_PI_MINUS * f[ifr] * px[ipx] * h[ih])
2625
# fast computation of exp(1j * x) - see https://stackoverflow.com/questions/9860711/cucomplex-h-and-exp/9863048#9863048
2726
s, c = cuda.libdevice.sincosf(TWO_PI_MINUS * f[ifr] * px[ipx] * h[ih])
28-
y[ih, ifr] += x[ipx, ifr] * (c + I * s)
27+
y[ih, ifr] += x[ipx, ifr] * (c + IMG * s)
2928

3029

3130
@cuda.jit
@@ -44,7 +43,7 @@ def _aradon_inner_2d_kernel(x, y, f, px, h, flim0, flim1, npx, nh):
4443
# x[ipx, ifr] += y[ih, ifr] * exp(TWO_PI_I_PLUS * f[ifr] * px[ipx] * h[ih])
4544
# fast computation of exp(1j * x) - see https://stackoverflow.com/questions/9860711/cucomplex-h-and-exp/9863048#9863048
4645
s, c = cuda.libdevice.sincosf(TWO_PI_PLUS * f[ifr] * px[ipx] * h[ih])
47-
x[ipx, ifr] += y[ih, ifr] * (c + I * s)
46+
x[ipx, ifr] += y[ih, ifr] * (c + IMG * s)
4847

4948

5049
def _radon_inner_2d_cuda(

pylops/signalprocessing/_fourierradon3d_cuda.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,11 @@
1-
from cmath import exp
21
from math import pi
32

43
import cupy as cp
54
from numba import cuda
65

76
TWO_PI_MINUS = cp.float32(-2.0 * pi)
87
TWO_PI_PLUS = cp.float32(2.0 * pi)
9-
I = cp.complex64(1j)
8+
IMG = cp.complex64(1j)
109

1110

1211
@cuda.jit
@@ -30,7 +29,7 @@ def _radon_inner_3d_kernel(x, y, f, py, px, hy, hx, flim0, flim1, npy, npx, nhy,
3029
s, c = cuda.libdevice.sincosf(
3130
TWO_PI_MINUS * f[ifr] * (py[ipy] * hy[ihy] + px[ipx] * hx[ihx])
3231
)
33-
y[ihy, ihx, ifr] += x[ipy, ipx, ifr] * (c + I * s)
32+
y[ihy, ihx, ifr] += x[ipy, ipx, ifr] * (c + IMG * s)
3433

3534

3635
@cuda.jit
@@ -53,7 +52,7 @@ def _aradon_inner_3d_kernel(x, y, f, py, px, hy, hx, flim0, flim1, npy, npx, nhy
5352
s, c = cuda.libdevice.sincosf(
5453
TWO_PI_PLUS * f[ifr] * (py[ipy] * hy[ihy] + px[ipx] * hx[ihx])
5554
)
56-
x[ipy, ipx, ifr] += y[ihy, ihx, ifr] * (c + I * s)
55+
x[ipy, ipx, ifr] += y[ihy, ihx, ifr] * (c + IMG * s)
5756

5857

5958
def _radon_inner_3d_cuda(

0 commit comments

Comments
 (0)