diff --git a/build/Android.bp b/build/Android.bp
index 6ab1d89..59717b8 100644
--- a/build/Android.bp
+++ b/build/Android.bp
@@ -50,6 +50,9 @@
     }) + select(release_flag("RELEASE_AVF_SUPPORT_CUSTOM_VM_WITH_PARAVIRTUALIZED_DEVICES"), {
         true: ["paravirtualized_devices"],
         default: [],
+    }) + select(release_flag("RELEASE_AVF_ENABLE_TPU_ASSIGNABLE_DEVICE"), {
+        true: ["tpu_assignable_device"],
+        default: [],
     }),
 }
 
@@ -64,6 +67,9 @@
     }) + select(release_flag("RELEASE_AVF_ENABLE_VIRT_CPUFREQ"), {
         true: ["-DAVF_ENABLE_VIRT_CPUFREQ=1"],
         default: [],
+    }) + select(release_flag("RELEASE_AVF_ENABLE_TPU_ASSIGNABLE_DEVICE"), {
+        true: ["-DAVF_ENABLE_TPU_ASSIGNABLE_DEVICE=1"],
+        default: [],
     }),
 }