@@ -397,7 +397,9 @@ def _fft(a, norm, out, forward, in_place, c2c, axes, batch_fft=True):
397
397
398
398
# cuFFT requires input arrays to be C-contiguous (row-major)
399
399
# for correct execution
400
- if dpnp .is_cuda_backend (a ) and not a .flags .c_contiguous :
400
+ if (
401
+ dpnp .is_cuda_backend (a ) and not a .flags .c_contiguous
402
+ ): # pragma: no cover
401
403
a = dpnp .ascontiguousarray (a )
402
404
403
405
# w/a for cuFFT to avoid "Invalid strides" error when
@@ -406,7 +408,7 @@ def _fft(a, norm, out, forward, in_place, c2c, axes, batch_fft=True):
406
408
# TODO: Remove this ones the OneMath issue is resolved
407
409
# https://github.com/uxlfoundation/oneMath/issues/631
408
410
cufft_wa = dpnp .is_cuda_backend (a ) and a .shape [- 1 ] == 1 and len (axes ) > 1
409
- if cufft_wa :
411
+ if cufft_wa : # pragma: no cover
410
412
a = dpnp .moveaxis (a , - 1 , - 2 )
411
413
412
414
a_strides = _standardize_strides_to_nonzero (a .strides , a .shape )
@@ -417,7 +419,7 @@ def _fft(a, norm, out, forward, in_place, c2c, axes, batch_fft=True):
417
419
res = _scale_result (res , a .shape , norm , forward , index )
418
420
419
421
# Revert swapped axes
420
- if cufft_wa :
422
+ if cufft_wa : # pragma: no cover
421
423
res = dpnp .moveaxis (res , - 1 , - 2 )
422
424
423
425
if batch_fft :
0 commit comments