diff --git a/buckbuild.bzl b/buckbuild.bzl index 3dd788ca28f..21ee68db4cf 100644 --- a/buckbuild.bzl +++ b/buckbuild.bzl @@ -739,7 +739,7 @@ def get_pt_operator_registry_dict( third_party("glog"), C10, ] + ([ROOT + ":torch_mobile_train"] if train else []) + - ([ROOT + ":torch_flatbuffer_all"] if enable_flatbuffer else []), + ([ROOT + ":flatbuffers_jit"] if enable_flatbuffer else []), **kwargs ) @@ -1347,7 +1347,7 @@ def define_buck_targets( exported_preprocessor_flags = get_pt_preprocessor_flags(), visibility = ["PUBLIC"], exported_deps = [ - ":torch_flatbuffer_all", + ":flatbuffers_jit", ":torch_mobile_core", ], ) @@ -1800,7 +1800,7 @@ def define_buck_targets( ) fb_xplat_cxx_library( - name = "torch_flatbuffer_all", + name = "flatbuffers_jit", visibility = ["PUBLIC"], exported_deps = [ ":flatbuffer_loader",