Testing hack

Author: BujSet

codegen/tools/gen_oplist.py

@@ -15,26 +15,10 @@
 
 try:
     from executorch.codegen.parse import strip_et_fields
-    from executorch.exir._serialize import _deserialize_pte_binary
-    from executorch.exir.schema import (
-        EValue,
-        KernelCall,
-        OptionalTensorList,
-        Tensor,
-        TensorList
-    )
 except ImportError:
     # If we build from source, executorch.codegen is not available.
     # We can use relative import instead.
     from ..parse import strip_et_fields
-    from ..exir._serialize import _deserialize_pte_binary
-    from ..exir.schema import (
-        EValue,
-        KernelCall,
-        OptionalTensorList,
-        Tensor,
-        TensorList
-    )
 
 from torchgen.gen import LineLoader, parse_native_yaml_struct
 from torchgen.selective_build.operator import SelectiveBuildOperator
@@ -105,6 +89,7 @@ def _get_operators(model_file: str) -> List[str]:
     with open(model_file, "rb") as f:
         buf = f.read()
 
+    from executorch.exir._serialize import _deserialize_pte_binary
     model = _deserialize_pte_binary(buf)
     operators = []
     for execution_plan in model.execution_plan:
@@ -114,19 +99,27 @@ def _get_operators(model_file: str) -> List[str]:
     return operators
 
 
-def _get_dtypes_from_non_list(evalue: EValue):
-    kernel_key = ""
-    if isinstance(evalue, Tensor):
-        dim_order = ",".join(map(str, evalue.dim_order))
-        kernel_key += f"{evalue.scalar_type};{dim_order}|"
-    return kernel_key
 
 
 def _get_kernel_metadata_for_model(model_file: str) -> Dict[str, List[str]]:
     with open(model_file, "rb") as f:
         buf = f.read()
     op_kernel_key_list: Dict[str, List[str]] = {}
 
+    from executorch.exir._serialize import _deserialize_pte_binary
+    from executorch.exir.schema import (
+        EValue,
+        KernelCall,
+        OptionalTensorList,
+        Tensor,
+        TensorList
+    )
+    def _get_dtypes_from_non_list(evalue: EValue):
+        kernel_key = ""
+        if isinstance(evalue, Tensor):
+            dim_order = ",".join(map(str, evalue.dim_order))
+            kernel_key += f"{evalue.scalar_type};{dim_order}|"
+        return kernel_key
     model = _deserialize_pte_binary(buf)
     for execution_plan in model.execution_plan:
         for chain in execution_plan.chains:

codegen/tools/targets.bzl

@@ -28,7 +28,7 @@ def define_common_targets(is_fbcode = False):
         main_module = "executorch.codegen.tools.gen_oplist",
         deps = [
             ":gen_oplist_lib",
-        ] + [] if runtime.is_oss else ["//executorch/exir:lib"],
+        ] + [] if runtime.is_oss else ["//executorch/exir:lib"], # TODO This is a hack to run on fbcode. These targets depends on exir which is only available in fbcode.
         preload_deps = [] if runtime.is_oss else ["//executorch/codegen/tools/fb:selective_build"],  # TODO(larryliu0820) :selective_build doesn't build in OSS yet
         package_style = "inplace",
         visibility = [