saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Break XPU] Align meta calculation for fft_r2c with _fft_r2c_mkl

Browse source 
Fix #146761

[ghstack-poisoned]
This commit is contained in:
xinan.lin 2025-02-08 08:20:36 -08:00
parent 926b4c4e5e
commit b0ce6ee974
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
torch/_meta_registrations.py
View file

@ -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)