frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add all bzl files per D36874458

Browse Source
This commit is contained in:
Jon Janzen
2022-06-03 14:19:26 -07:00
parent 1f53d036d2
commit 03847808a0
34 changed files with 3888 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
android/build_defs.bzl Normal file
View File

@ -0,0 +1,19 @@
load("@fbsource//tools/build_defs:fb_xplat_cxx_test.bzl", "fb_xplat_cxx_test")
load("@fbsource//xplat/caffe2:pt_defs.bzl", "get_build_from_deps_query", "pt_operator_registry")
DEFAULT_PT_OP_DEPS = [
"fbsource//xplat/caffe2:torch_mobile_ops_full_dev",
]
def pt_xplat_cxx_test(name, deps = [], pt_op_deps = DEFAULT_PT_OP_DEPS, **kwargs):
code_gen_lib = []
if get_build_from_deps_query():
lib_name = name + "_lib"
pt_operator_registry(lib_name, preferred_linkage = "static", template_select = False, deps = pt_op_deps)
code_gen_lib = [":" + lib_name]
deps = deps + code_gen_lib
fb_xplat_cxx_test(
name = name,
deps = deps,
**kwargs
)