Update base for Update on "[Executorch][llm] Compile custom op with -O2"

This enable some optimizations for inlining vectorized lib functions

Differential Revision: [D64910576](https://our.internmc.facebook.com/intern/diff/D64910576/)

[ghstack-poisoned]

Author: kimishpatel

kernels/optimized/lib_defs.bzl

@@ -2,8 +2,6 @@ load("@fbsource//tools/build_defs:default_platform_defs.bzl", "DEVSERVER_PLATFOR
 load("@fbsource//tools/build_defs:fb_native_wrapper.bzl", "fb_native")
 load("@fbsource//xplat/executorch/backends/xnnpack/third-party:third_party_libs.bzl", "third_party_dep")
 load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
-lobd("@fbsource//tools/build_defs/apple/build_mode_defs.bzl", "is_local_build")
-lobd("@fbsource//tools/build_defs/android/build_mode_defs.bzl", "is_production_build")
 
 # Because vec exists as a collection of header files, compile and preprocessor
 # flags applied to the vec target do not have any effect, since no compilation
@@ -123,8 +121,21 @@ def define_libs():
             exported_headers = native.glob([
                 "blas/**/*.h",
             ]),
-            fbobjc_compiler_flags = [] if is_local_build() else ["-O2"],
-            fbandroid_compiler_flags = ["-O2"] if is_production_build() else [],
+            compiler_flags = select({
+              "DEFAULT": [],
+              "ovr_config//os:android-arm64": [
+                    "-O2",
+              ] if not runtime.is_oss else [],
+              "ovr_config//os:iphoneos": [
+                  "-O2",
+              ] if not runtime.is_oss else [],
+              "ovr_config//os:macos-arm64": [
+                  "-O2",
+              ] if not runtime.is_oss else [],
+              "ovr_config//os:macos-x86_64": [
+                  "-O2",
+              ] if not runtime.is_oss else [],
+            }),
             header_namespace = "executorch/kernels/optimized",
             visibility = [
                 "//executorch/...",

kernels/optimized/op_registration_util.bzl

@@ -87,7 +87,21 @@ def define_op_library(name, deps):
         ],
         # kernels often have helpers with no prototypes just disabling the warning here as the headers
         # are codegend and linked in later
-        compiler_flags = ["-Wno-missing-prototypes", "-O2"],
+        compiler_flags = ["-Wno-missing-prototypes"] + select({
+          "DEFAULT": [],
+          "ovr_config//os:android": [
+                "-O2",
+          ] if not runtime.is_oss else [],
+          "ovr_config//os:iphoneos": [
+              "-O2",
+          ] if not runtime.is_oss else [],
+          "ovr_config//os:macos-arm64": [
+              "-O2",
+          ] if not runtime.is_oss else [],
+          "ovr_config//os:macos-x86_64": [
+              "-O2",
+          ] if not runtime.is_oss else [],
+        }),
         deps = [
             "//executorch/runtime/kernel:kernel_includes",
         ] + augmented_deps,