Reapply "Add (and fix) OSS check that et_operator_library dtype_selective_build builds (#9245)" (#11589)

This PR was created by the merge bot to help merge the original PR into
the main branch.
ghstack PR number: #11573 by
@swolchok
^ Please use this as the source of truth for the PR details, comments,
and reviews
ghstack PR base:
https://github.com/pytorch/executorch/tree/gh/swolchok/453/base
ghstack PR head:
https://github.com/pytorch/executorch/tree/gh/swolchok/453/head
Merge bot PR base: https://github.com/pytorch/executorch/tree/main
Merge bot PR head:
https://github.com/pytorch/executorch/tree/gh/swolchok/453/orig
@diff-train-skip-merge

Co-authored-by: Scott Wolchok <[email protected]>

Author: pytorchbot

.ci/scripts/unittest-buck2.sh

@@ -25,7 +25,9 @@ BUILDABLE_KERNELS_PRIM_OPS_TARGETS=$(buck2 query //kernels/prim_ops/... | grep -
 # //runtime/kernel/... is failing because //third-party:torchgen_files's shell script can't find python on PATH.
 # //runtime/test/... requires Python torch, which we don't have in our OSS buck setup.
 for op in "build" "test"; do
-    buck2 $op $BUILDABLE_OPTIMIZED_OPS //kernels/portable/... \
+    buck2 $op $BUILDABLE_OPTIMIZED_OPS \
+          //examples/selective_build:select_all_dtype_selective_lib_portable_lib \
+          //kernels/portable/... \
           $BUILDABLE_KERNELS_PRIM_OPS_TARGETS //runtime/backend/... //runtime/core/... \
           //runtime/executor: //runtime/kernel/... //runtime/platform/...
 done

examples/selective_build/targets.bzl

@@ -25,6 +25,22 @@ def define_common_targets():
         ],
     )
 
+    if runtime.is_oss or is_xplat():
+        executorch_generated_lib(
+            name = "select_all_dtype_selective_lib",
+            functions_yaml_target = "//executorch/kernels/portable:functions.yaml",
+            kernel_deps = [
+                "//executorch/kernels/portable:operators",
+            ],
+            # Setting dtype_selective_build without using list or dict selection isn't a
+            # typical use case; we just do it here so that we can test that our mechanism
+            # for getting buck deps right for dtype_selective_build is working.
+            dtype_selective_build = True,
+            deps = [
+                ":select_all_ops",
+            ],
+        )
+
     # Select a list of operators: defined in `ops`
     et_operator_library(
         name = "select_ops_in_list",

shim_et/xplat/executorch/codegen/codegen.bzl

@@ -752,7 +752,7 @@ def executorch_generated_lib(
             See: https://www.internalfb.com/wiki/PyTorch/Teams/Edge/PyTorch_Edge_Core_Team/Dtype_Selective_Build/""")
 
     if dtype_selective_build:
-        if not expose_operator_symbols and not is_xplat():
+        if not expose_operator_symbols and not (is_xplat() or runtime.is_oss):
             fail("""
                 Dtype selective build with expose_operator_symbols=False works only in xplat -
                 there are undefined symbols otherwise. Please try to use xplat, or talk to the