Experimental

Author: kirklandsign

extension/android/CMakeLists.txt

@@ -170,8 +170,8 @@ if(EXECUTORCH_BUILD_EXTENSION_TRAINING)
 endif()
 
 if(EXECUTORCH_BUILD_LLAMA_JNI)
-  target_sources(executorch_jni PRIVATE jni/jni_layer_llama.cpp jni/log.cpp)
-  list(APPEND link_libraries extension_llm_runner)
+  target_sources(executorch_jni PRIVATE jni/jni_layer_llama.cpp jni/jni_layer_asr.cpp jni/log.cpp)
+  list(APPEND link_libraries extension_llm_runner extension_asr_runner)
   target_compile_definitions(executorch_jni PUBLIC EXECUTORCH_BUILD_LLAMA_JNI=1)
 
   if(QNN_SDK_ROOT)
@@ -224,11 +224,6 @@ if(EXECUTORCH_BUILD_LLAMA_JNI)
   endif()
 endif()
 
-if(EXECUTORCH_BUILD_ASR_JNI)
-  target_sources(executorch_jni PRIVATE jni/jni_layer_asr.cpp)
-  list(APPEND link_libraries extension_asr_runner)
-endif()
-
 target_include_directories(
   executorch_jni
   PRIVATE

extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/asr/AsrCallback.kt

@@ -8,13 +8,16 @@
 
 package org.pytorch.executorch.extension.asr
 
+import org.pytorch.executorch.annotations.Experimental
+
 /**
  * Callback interface for ASR (Automatic Speech Recognition) module.
  * Users can implement this interface to receive the transcribed tokens
  * and completion notification.
  *
  * Warning: These APIs are experimental and subject to change without notice
  */
+@Experimental
 interface AsrCallback {
     /**
      * Called when a new token is available from JNI.

extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/asr/AsrModule.kt

@@ -10,6 +10,7 @@ package org.pytorch.executorch.extension.asr
 
 import java.io.Closeable
 import java.io.File
+import org.pytorch.executorch.annotations.Experimental
 
 /**
  * AsrModule is a wrapper around the ExecuTorch ASR Runner. It provides a simple interface to
@@ -24,6 +25,7 @@ import java.io.File
  * @param tokenizerPath Path to the tokenizer file
  * @param dataPath Optional path to additional data file (e.g., for delegate data)
  */
+@Experimental
 class AsrModule(
     modelPath: String,
     tokenizerPath: String,

extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/asr/AsrTranscribeConfig.kt

@@ -8,6 +8,8 @@
 
 package org.pytorch.executorch.extension.asr
 
+import org.pytorch.executorch.annotations.Experimental
+
 /**
  * Configuration for ASR transcription.
  *
@@ -17,6 +19,7 @@ package org.pytorch.executorch.extension.asr
  * @property temperature Temperature for sampling. 0.0 means greedy decoding
  * @property decoderStartTokenId The token ID to start decoding with (e.g., language token for Whisper)
  */
+@Experimental
 data class AsrTranscribeConfig(
     val maxNewTokens: Long = 128,
     val temperature: Float = 0.0f,