Aborted (core dumped) in tf.irfftnd/ifftnd/irfftnd

[x0w3n]

Issue type

Bug

Have you reproduced the bug with TensorFlow Nightly?

No

Source

source

TensorFlow version

tf 2.16

Custom code

Yes

OS platform and distribution

Linux Ubuntu 22.04.3 LTS (x86_64)

Mobile device

No response

Python version

3.9.13

Bazel version

No response

GCC/compiler version

No response

CUDA/cuDNN version

No response

GPU model and memory

No response

Current behavior?

When the fft_length parameter of tf.irfftnd/ifftnd/irfftnd is less than or equal to 0, it will trigger a crash, resulting in Check failed.

Standalone code to reproduce the issue

import tensorflow as tf

input1 = tf.random.normal([10,10,10,10,10], dtype=tf.float32)
input2 = tf.random.normal([10,10,10,10,10], dtype=tf.float32)
input3 = tf.complex(input1,input2)

tf.irfftnd(input3,-3,-3) #crash
# tf.ifftnd(input3,-3,-3) #crash
# tf.irfftnd(input3,-3,-3) #crash

Relevant log output

2024-05-07 07:46:42.396021: F tensorflow/core/framework/tensor_shape.cc:45] Check failed: NDIMS == dims() (1 vs. 0)Asking for tensor of 1 dimensions from a tensor of 0 dimensions
Aborted (core dumped)

[Venkat6871]

Hi @x0w3n ,

Sorry for the delay, I tried to run your code on colab using TF v2.16, nightly and faced the same issue. Please find the gist here for reference.

Thank you!

[x0w3n]

Hi @x0w3n ,

Sorry for the delay, I tried to run your code on colab using TF v2.16, nightly and faced the same issue. Please find the gist here for reference.

Thank you!
OK. Thank you!