mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
[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:
committed by
PyTorch MergeBot
parent
5b88a2078b
commit
0b3a239e85
10
third_party/BUCK.oss
vendored
10
third_party/BUCK.oss
vendored
@ -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 = "",
|
||||
|
Reference in New Issue
Block a user