Add a latest XNNPACK opt-in (pytorch#22972)

Summary:

Keep ExecuTorch on the existing XNNPACK dependency by default and add
`-c executorch.use_latest_xnnpack=true` to opt in

fb:

Reviewed By: digantdesai

Differential Revision: D116485510

Pull Request resolved: pytorch#22972

Author: JakeStevens

backends/vulkan/test/op_tests/targets.bzl

@@ -1,6 +1,7 @@
 load("@fbsource//tools/build_defs:platform_defs.bzl", "ANDROID")
 load("@fbsource//xplat/caffe2:pt_defs.bzl", "get_pt_ops_deps")
 load("@fbsource//xplat/caffe2:pt_ops.bzl", "pt_operator_library")
+load("@fbsource//xplat/executorch/backends/xnnpack/third-party:third_party_libs.bzl", "third_party_dep")
 load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
 load(
     "@fbsource//xplat/executorch/backends/vulkan:targets.bzl",
@@ -232,6 +233,6 @@ def define_common_targets(is_fbcode = False):
         "conv2d_gemm_dynamic_test",
         extra_deps = [
             "//executorch/runtime/core/exec_aten/testing_util:tensor_util",
-            "fbsource//xplat/third-party/XNNPACK:XNNPACK",
+            third_party_dep("XNNPACK"),
         ]
     )

shim_et/xplat/executorch/backends/xnnpack/third-party/third_party_libs.bzl

@@ -1,5 +1,7 @@
 load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
 
+_XNNPACK_LATEST = "fbsource//third-party/XNNPACK:XNNPACK"
+
 # 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
@@ -17,4 +19,8 @@ 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]
+    if runtime.is_oss:
+        return _THIRD_PARTY_LIBS[name][1]
+    if name == "XNNPACK" and native.read_config("executorch", "use_latest_xnnpack", "false") == "true":
+        return _XNNPACK_LATEST
+    return _THIRD_PARTY_LIBS[name][0]