Merge "protos: add descriptor set build import"
diff --git a/protos/BUILD b/protos/BUILD
index dc8be4e..34fabf1 100644
--- a/protos/BUILD
+++ b/protos/BUILD
@@ -15,6 +15,7 @@
 # This file is automatically generated by tools/gen_bazel. Do not edit.
 
 load("//tools/build_defs/proto/cpp:cc_proto_library.bzl", "cc_proto_library")
+load("//tools/build_defs/proto:descriptor_set.bzl", "transitive_descriptor_set")
 load("//third_party/perfetto/google:build_defs.bzl", "pbzero_cc_proto_library")
 
 package(default_visibility = ["//third_party/perfetto:__subpackages__"])
diff --git a/tools/gen_bazel b/tools/gen_bazel
index 1ed0ff7..51a1e68 100755
--- a/tools/gen_bazel
+++ b/tools/gen_bazel
@@ -327,9 +327,14 @@
     self.build = Build(public=True)
 
     proto_cc_import = 'load("//tools/build_defs/proto/cpp:cc_proto_library.bzl", "cc_proto_library")'
+    descriptor_set_import = 'load("//tools/build_defs/proto:descriptor_set.bzl", "transitive_descriptor_set")'
     pbzero_cc_import = 'load("//third_party/perfetto/google:build_defs.bzl", "pbzero_cc_proto_library")'
-    self.proto_build = Build(public=False, header_lines=[
-                        proto_cc_import, pbzero_cc_import])
+    self.proto_build = Build(public=False,
+                             header_lines=[
+                               proto_cc_import,
+                               descriptor_set_import,
+                               pbzero_cc_import,
+                             ])
 
     for target in targets:
       self.create_target(target)