1*523fa7a6SAndroid Build Coastguard Worker# Any targets that should be shared between fbcode and xplat must be defined in 2*523fa7a6SAndroid Build Coastguard Worker# targets.bzl. This file can contain fbcode-only targets. 3*523fa7a6SAndroid Build Coastguard Worker 4*523fa7a6SAndroid Build Coastguard Workerload("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime") 5*523fa7a6SAndroid Build Coastguard Workerload(":targets.bzl", "define_common_targets") 6*523fa7a6SAndroid Build Coastguard Worker 7*523fa7a6SAndroid Build Coastguard Workeroncall("executorch") 8*523fa7a6SAndroid Build Coastguard Worker 9*523fa7a6SAndroid Build Coastguard Workerdefine_common_targets() 10*523fa7a6SAndroid Build Coastguard Worker 11*523fa7a6SAndroid Build Coastguard Workerruntime.cxx_python_extension( 12*523fa7a6SAndroid Build Coastguard Worker name = "_training_lib", 13*523fa7a6SAndroid Build Coastguard Worker srcs = [ 14*523fa7a6SAndroid Build Coastguard Worker "_training_lib.cpp", 15*523fa7a6SAndroid Build Coastguard Worker ], 16*523fa7a6SAndroid Build Coastguard Worker base_module = "executorch.extension.training.pybindings", 17*523fa7a6SAndroid Build Coastguard Worker types = ["_training_lib.pyi"], 18*523fa7a6SAndroid Build Coastguard Worker visibility = ["//executorch/extension/training/..."], 19*523fa7a6SAndroid Build Coastguard Worker deps = [ 20*523fa7a6SAndroid Build Coastguard Worker "//executorch/extension/aten_util:aten_bridge", 21*523fa7a6SAndroid Build Coastguard Worker "//executorch/extension/training/optimizer:sgd", 22*523fa7a6SAndroid Build Coastguard Worker ], 23*523fa7a6SAndroid Build Coastguard Worker external_deps = [ 24*523fa7a6SAndroid Build Coastguard Worker "pybind11", 25*523fa7a6SAndroid Build Coastguard Worker "libtorch_python", 26*523fa7a6SAndroid Build Coastguard Worker ], 27*523fa7a6SAndroid Build Coastguard Worker) 28*523fa7a6SAndroid Build Coastguard Worker 29*523fa7a6SAndroid Build Coastguard Workerruntime.python_library( 30*523fa7a6SAndroid Build Coastguard Worker name = "_training_module", 31*523fa7a6SAndroid Build Coastguard Worker srcs = [ 32*523fa7a6SAndroid Build Coastguard Worker "_training_module.py", 33*523fa7a6SAndroid Build Coastguard Worker ], 34*523fa7a6SAndroid Build Coastguard Worker base_module = "executorch.extension.training.pybindings", 35*523fa7a6SAndroid Build Coastguard Worker visibility = ["//executorch/extension/training/..."], 36*523fa7a6SAndroid Build Coastguard Worker deps = [ 37*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 38*523fa7a6SAndroid Build Coastguard Worker "//executorch/extension/pybindings:portable_lib", 39*523fa7a6SAndroid Build Coastguard Worker ], 40*523fa7a6SAndroid Build Coastguard Worker) 41