map third party deps between OSS and Internal

Summary:
In order for the XNNPACK delegate's buck targets to work both internally and for OSS, we need to be able to switch the third-party XNNPACK dependency depending on whether or not we are building for OSS or for Internal use.

We have something similar in PyTorch, so we are doing something similar here. We disentangle this from ExecuTorch runtime wrapper in order to separate the XNNPACK delegate's third-party dependency from Core Executorch's Third-Party Dependencies.

Reviewed By: dbort

Differential Revision: D48214072

fbshipit-source-id: f12505f6b96a4bf7066a46f2f9035cc4ce64f3b6

Author: mcr229

backends/xnnpack/targets.bzl

@@ -1,3 +1,4 @@
+load("@fbsource//xplat/executorch/backends/xnnpack/third-party:third_party_libs.bzl", "third_party_dep")
 load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
 
 def define_common_targets():
@@ -49,7 +50,7 @@ def define_common_targets():
             "@EXECUTORCH_CLIENTS",
         ],
         deps = [
-            "//xplat/third-party/XNNPACK:XNNPACK",
+            third_party_dep("XNNPACK"),
             ":xnnpack_schema",
             "//executorch/runtime/backend:backend_registry",
             "//executorch/backends/qnnpack:qnnpack_utils",  # TODO Use (1) portable for choose_qparams(), (2) xnnpack for quantize_per_tensor()

backends/xnnpack/third-party/third_party_libs.bzl

@@ -0,0 +1,20 @@
+load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
+
+# Dictionary mappping third-party library name to the correct OSS/Internal target dependencies.
+# the values of the dictionary are lists where the first element is the internal dep and the
+# second element is the OSS dep
+_THIRD_PARTY_LIBS = {
+    "FP16": ["//xplat/third-party/FP16:FP16", "//backends/xnnpack/third-party:FP16"],
+    "FXdiv": ["//xplat/third-party/FXdiv:FXdiv", "//backends/xnnpack/third-party:FXdiv"],
+    "XNNPACK": ["//xplat/third-party/XNNPACK:XNNPACK", "//backends/xnnpack/third-party:XNNPACK"],
+    "clog": ["//xplat/third-party/clog:clog", "//backends/xnnpack/third-party:clog"],
+    "cpuinfo": ["//third-party/cpuinfo:cpuinfo", "//backends/xnnpack/third-party:cpuinfo"],
+    "pthreadpool": ["//xplat/third-party/pthreadpool:pthreadpool", "//backends/xnnpack/third-party:pthreadpool"],
+    "pthreadpool_header": ["//xplat/third-party/pthreadpool:pthreadpool_header", "//backends/xnnpack/third-party:pthreadpool_header"],
+}
+
+def third_party_dep(name):
+    if name not in _THIRD_PARTY_LIBS:
+        fail("Cannot find third party library " + name + ", please register it in THIRD_PARTY_LIBS first!")
+
+    return _THIRD_PARTY_LIBS[name][1] if runtime.is_oss else _THIRD_PARTY_LIBS[name][0]