Pybindings Module is more flexible with non_const set ups (#109)

Summary:
Pull Request resolved: #109

Made the module handle models with more then one non_const buffer level, and made it allocate the needed memory dynamically rather then relying on static constants which historically have needed to be updated a ton.

Reviewed By: guangy10

Differential Revision: D48583639

fbshipit-source-id: bd39606cf818e12f427b94a53162f4c16df4db73

Author: JacobSzwejbka

extension/pybindings/TARGETS

@@ -2,7 +2,7 @@
 # Any targets that should be shared between fbcode and xplat must be defined in
 # targets.bzl. This file can contain fbcode-only targets.
 
-load("@fbcode//executorch/extension/pybindings:targets.bzl", "MODELS_ATEN_OPS_ATEN_MODE_GENERATED_LIB", "MODELS_ATEN_OPS_LEAN_MODE_GENERATED_LIB", "MODULE_DEPS", "define_common_targets", "executorch_pybindings")
+load("@fbcode//executorch/extension/pybindings:targets.bzl", "ATEN_MODULE_DEPS", "MODELS_ATEN_OPS_ATEN_MODE_GENERATED_LIB", "MODELS_ATEN_OPS_LEAN_MODE_GENERATED_LIB", "PORTABLE_MODULE_DEPS", "define_common_targets", "executorch_pybindings")
 load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
 
 define_common_targets()
@@ -11,7 +11,7 @@ executorch_pybindings(
     srcs = [
         "module.cpp",
     ],
-    cppdeps = MODULE_DEPS,
+    cppdeps = PORTABLE_MODULE_DEPS,
     python_module_name = "core",
     visibility = ["PUBLIC"],
 )
@@ -20,7 +20,7 @@ executorch_pybindings(
     srcs = [
         "module.cpp",
     ],
-    cppdeps = MODULE_DEPS + MODELS_ATEN_OPS_LEAN_MODE_GENERATED_LIB,
+    cppdeps = PORTABLE_MODULE_DEPS + MODELS_ATEN_OPS_LEAN_MODE_GENERATED_LIB,
     python_module_name = "portable",
     visibility = ["PUBLIC"],
 )
@@ -29,20 +29,7 @@ executorch_pybindings(
     srcs = [
         "module.cpp",
     ],
-    cppdeps = [
-        "//executorch/runtime/kernel:operator_registry",
-        "//executorch/runtime/executor:executor_aten",
-        "//executorch/runtime/core/exec_aten:lib",
-        "//executorch/schema:bundled_program_schema",
-        "//executorch/schema:program",
-        "//executorch/extension/data_loader:buffer_data_loader",
-        "//executorch/extension/data_loader:mmap_data_loader",
-        "//executorch/util:read_file",
-        "//executorch/util:test_memory_config",
-        "//executorch/util:bundled_program_verification_aten",
-        "//caffe2:torch-cpp",
-        "//executorch/runtime/executor/test:test_backend_compiler_lib_aten",
-    ] + MODELS_ATEN_OPS_ATEN_MODE_GENERATED_LIB,
+    cppdeps = ATEN_MODULE_DEPS + MODELS_ATEN_OPS_ATEN_MODE_GENERATED_LIB,
     python_module_name = "aten_mode_lib",
     visibility = ["PUBLIC"],
 )