Fail to use jittor.nn._fft2()

[Joey1998hub]

Describe the bug

Fail to use nn._fft2() After nn._fft2(), the data type has been changed from jt.Var to jt.jittor_core.Var, the log shows some errors during compiling.

Full Log

Minimal Reproduce

import jittor as jt import numpy as np from jittor import nn

jt.flags.use_cuda = 1 def test_fft_forward(): img = np.random.rand(256, 300) img2 = np.random.rand(256, 300) X = np.stack([img, img2], 0) x = jt.array(X,dtype=jt.float32) x = jt.stack([x, jt.zeros_like(x)], 3) y = nn._fft2(x) y_jt_real = y[:, :, :, 0].data y_jt_imag = y[:, :, :, 1].data

test_fft_forward()

Expected behavior

Hope the problem can be solved and jt.nn._fft2() can be executed correctly and quickly.

[Jittor]

Thanks for reporting, we are working on it.

[cxjyxxme]

This problem is fixed in the latest version of jittor, please upgrade your jittor.