Buckify Llama multimodal export

Summary: Buckify export path for Llama multimodal

Reviewed By: iseeyuan

Differential Revision: D67949092

Author: jackzhxng

examples/models/llama/TARGETS

@@ -107,6 +107,7 @@ runtime.python_library(
         "//ai_codesign/gen_ai/fast_hadamard_transform:fast_hadamard_transform",
         "//caffe2:torch",
         "//executorch/backends/vulkan/_passes:vulkan_passes",
+        "//executorch/exir/passes:init_mutable_pass",
         "//executorch/examples/models:model_base",
         "//executorch/examples/models:models",
         "//executorch/extension/llm/custom_ops:custom_ops_aot_py",

examples/models/llama3_2_vision/TARGETS

@@ -0,0 +1,14 @@
+load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
+
+oncall("executorch")
+
+python_library(
+    name = "multimodal_lib",
+    srcs = [
+        "__init__.py",
+    ],
+    deps = [
+        "//executorch/examples/models/llama3_2_vision/text_decoder:model",
+        "//executorch/examples/models/llama3_2_vision/vision_encoder:model",
+    ],
+)

examples/models/llama3_2_vision/text_decoder/TARGETS

@@ -0,0 +1,17 @@
+load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
+
+oncall("executorch")
+
+python_library(
+    name = "model",
+    srcs = [
+        "model.py",
+    ],
+    deps = [
+        "//caffe2:torch",
+        "//executorch/examples/models:checkpoint",
+        "//pytorch/torchtune:lib",
+        "//executorch/extension/llm/modules:module_lib",
+    ],
+)
+ 

examples/models/llama3_2_vision/vision_encoder/TARGETS

@@ -0,0 +1,17 @@
+load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
+
+oncall("executorch")
+
+python_library(
+    name = "model",
+    srcs = [
+        "__init__.py",
+        "model.py",
+    ],
+    deps = [
+        "//caffe2:torch",
+        "//executorch/extension/llm/modules:module_lib",
+        "//pytorch/torchtune:lib",
+        "//executorch/examples/models:model_base",
+    ],
+)

exir/passes/TARGETS

@@ -220,6 +220,16 @@ python_library(
     ],
 )
 
+python_library(
+    name = "init_mutable_pass",
+    srcs = [
+        "init_mutable_pass.py",
+    ],
+    deps = [
+        "//executorch/exir:delegate",
+    ],
+)
+
 python_library(
     name = "sym_shape_eval_pass",
     srcs = [

extension/llm/modules/TARGETS

@@ -0,0 +1,49 @@
+load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
+
+oncall("executorch")
+
+python_library(
+    name = "kv_cache",
+    srcs = [
+        "kv_cache.py",
+    ],
+    deps = [
+        "//caffe2:torch",
+        "//pytorch/torchtune:lib",
+    ],
+)
+
+python_library(
+    name = "attention",
+    srcs = [
+        "attention.py",
+    ],
+    deps = [
+        ":kv_cache",
+        "//caffe2:torch",
+        "//executorch/extension/llm/custom_ops:custom_ops",
+        "//pytorch/torchtune:lib",
+    ],
+)
+
+python_library(
+    name = "position_embeddings",
+    srcs = [
+        "_position_embeddings.py",
+    ],
+    deps = [
+        "//caffe2:torch",
+    ],
+)
+
+python_library(
+    name = "module_lib",
+    srcs = [
+        "__init__.py",
+    ],
+    deps= [
+        ":position_embeddings",
+        ":attention",
+        ":kv_cache",
+    ]
+)