[pocket fft] turning on pocketfft flag (#81670)

Summary:
enabling AT_POCKETFFT_ENABLED@ flag and adding the appropriate dependencies to aten-cpu

moved mkl files from
`aten_cpu_source_non_codegen_list` to
`aten_native_source_non_codegen_list`

Test Plan:
After building testing binaries for both android and ios targets

### iOS
`fbcode/aibench/specifications/frameworks/pytorch/ios/build.sh`

Submitted benchmarks with the new binaries supporting pocketfft here:
https://www.internalfb.com/intern/aibench/details/245253003946591

### Android
`fbcode/aibench/specifications/frameworks/pytorch/android/arm64/build.sh`

Submitted Benchmarks with the new binaries supporting pocket fft here:
https://www.internalfb.com/intern/aibench/details/406253690682941

### Build Size Impact

Success: igios-pika on D37790257-V7

☷[pocket fft] turning on pocketfft flag☷
Diff: https://fburl.com/diff/exkploof
Unigraph Explorer: https://fburl.com/mbex/aipdzaqo

Changes for variation [arm64 + 3x assets]:
```Compressed  : -473 B (-0.00%) => 86.69 MiB
Uncompressed: +2.4 KiB (+0.00%) => 187.71 MiB
```

Reviewed By: kimishpatel

Differential Revision: D37790257

Pull Request resolved: https://github.com/pytorch/pytorch/pull/81670
Approved by: https://github.com/kit1980
This commit is contained in:
Max Ren
2022-07-21 02:45:20 +00:00
committed by PyTorch MergeBot
parent 5b88a2078b
commit 0b3a239e85
3 changed files with 17 additions and 5 deletions

10
third_party/BUCK.oss vendored
View File

@ -75,6 +75,16 @@ cxx_library(
visibility = ["PUBLIC"],
)
cxx_library(
name = "pocketfft_header",
header_namespace = "",
exported_headers = {
"pocketfft_hdronly.h": "pocketfft/pocketfft_hdronly.h",
},
reexport_all_header_dependencies = True,
visibility = ["PUBLIC"],
)
cxx_library(
name = "FXdiv",
header_namespace = "",