From b0ce6ee9743dbda21da3fb12026307aef093414f Mon Sep 17 00:00:00 2001 From: "xinan.lin" Date: Sat, 8 Feb 2025 08:20:36 -0800 Subject: [PATCH] [Break XPU] Align meta calculation for fft_r2c with _fft_r2c_mkl Fix #146761 [ghstack-poisoned] --- torch/_meta_registrations.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/torch/_meta_registrations.py b/torch/_meta_registrations.py index c1ef520c967..b489ba12fb2 100644 --- a/torch/_meta_registrations.py +++ b/torch/_meta_registrations.py @@ -370,9 +370,11 @@ def meta_fft_r2c(self, dim, normalization, onesided): return output else: - return self.new_empty( + sorted_dims = _sort_dims(self, dim, exclude_last=True) + out = self.new_empty( out_sizes, dtype=utils.corresponding_complex_dtype(self.dtype) ) + return _exec_fft(out, self, out_sizes, sorted_dims, forward=True) @register_meta(aten.randperm.generator_out)