Allow setting exec_group in more places

Signed-off-by: Brentley Jones <[email protected]>

Author: brentleyjones

proto/swift_proto_utils.bzl

@@ -196,6 +196,7 @@ def compile_swift_protos_for_target(
         additional_swift_proto_compiler_info,
         attr,
         ctx,
+        exec_group = None,
         module_name,
         proto_infos,
         swift_proto_compilers,
@@ -208,6 +209,8 @@ def compile_swift_protos_for_target(
         additional_compiler_deps: Additional dependencies passed directly to the Swift compiler.
         attr: The attributes of the target for which the module is being compiled.
         ctx: The context of the aspect or rule.
+        exec_group: Runs the Swift compilation action under the given execution
+            group's context. If `None`, the default execution group is used.
         module_name: The name of the Swift module that should be compiled from the protos.
         proto_infos: List of `ProtoInfo` providers to compile into Swift source files.
         swift_proto_compilers: List of targets propagating `SwiftProtoCompiler` providers.
@@ -276,6 +279,7 @@ def compile_swift_protos_for_target(
         additional_inputs = additional_inputs,
         cc_infos = get_providers(compiler_deps, CcInfo),
         copts = ["-parse-as-library"] + copts,
+        exec_group = exec_group,
         feature_configuration = feature_configuration,
         include_dev_srch_paths = include_dev_srch_paths,
         module_name = module_name,

swift/internal/symbol_graph_extracting.bzl

@@ -25,6 +25,7 @@ def extract_symbol_graph(
         actions,
         compilation_contexts,
         emit_extension_block_symbols = None,
+        exec_group = None,
         feature_configuration,
         include_dev_srch_paths,
         minimum_access_level = None,
@@ -44,6 +45,8 @@ def extract_symbol_graph(
             a target's dependencies.
         emit_extension_block_symbols: A `bool` that indicates whether `extension` block
             information should be included in the symbol graph.
+        exec_group: Runs the Swift compilation action under the given execution
+            group's context. If `None`, the default execution group is used.
         feature_configuration: The Swift feature configuration.
         include_dev_srch_paths: A `bool` that indicates whether the developer
             framework search paths will be added to the compilation command.
@@ -114,6 +117,7 @@ def extract_symbol_graph(
     run_toolchain_action(
         actions = actions,
         action_name = SWIFT_ACTION_SYMBOL_GRAPH_EXTRACT,
+        exec_group = exec_group,
         feature_configuration = feature_configuration,
         outputs = [output_dir],
         prerequisites = prerequisites,