OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Move xnnpack taget to fb code base (#88909)

Browse Source 
1. Move the source file list to the `build_variables.bzl`, as it's the source of truth for both internal buck build and oss build
2. Move target definitions to `fb` internal folder
3. Some changes are triggered from auto format.

Differential Revision: [D40906961](https://our.internmc.facebook.com/intern/diff/D40906961/)

**NOTE FOR REVIEWERS**: This PR has internal Meta-specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D40906961/)!
Pull Request resolved: https://github.com/pytorch/pytorch/pull/88909
Approved by: https://github.com/mcr229
This commit is contained in:
Chen Lai 2022-11-12 21:41:31 -08:00 committed by PyTorch MergeBot
parent 2b12bfce88
commit bca75fd2d3
1 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
build_variables.bzl
View File

@ -1489,3 +1489,33 @@ aten_cuda_with_sort_by_key_source_list = [
aten_cuda_cu_with_sort_by_key_source_list = [
"aten/src/ATen/native/cuda/Unique.cu",
]
# Followings are source code for xnnpack delegate
xnnpack_delegate_serializer_header = [
"torch/csrc/jit/backends/xnnpack/serialization/serializer.h",
]
xnnpack_delegate_serializer_source_list = [
"torch/csrc/jit/backends/xnnpack/serialization/serializer.cpp",
]
xnnpack_delegate_core_source_list = [
"torch/csrc/jit/backends/xnnpack/compiler/xnn_compiler.cpp",
]
xnnpack_delegate_core_header = [
"torch/csrc/jit/backends/xnnpack/compiler/xnn_compiler.h",
"torch/csrc/jit/backends/xnnpack/executor/xnn_executor.h",
]
xnnpack_backend_header = [
"torch/csrc/jit/backends/xnnpack/xnnpack_graph_builder.h",
] + xnnpack_delegate_core_header
xnnpack_backend_source_list = [
"torch/csrc/jit/backends/xnnpack/compiler/xnn_compiler.cpp",
"torch/csrc/jit/backends/xnnpack/xnnpack_backend_lib.cpp",
"torch/csrc/jit/backends/xnnpack/xnnpack_backend_preprocess.cpp",
"torch/csrc/jit/backends/xnnpack/xnnpack_graph_builder.cpp",
] + xnnpack_delegate_core_source_list