From aa203297d4074ff9e11be08871fd3ca412d65ff7 Mon Sep 17 00:00:00 2001 From: Michal Harakal Date: Fri, 4 Sep 2026 22:27:51 +0200 Subject: [PATCH] chore(api): refresh functiongemma/gemma dumps for #415/#417's new API surface forwardAt, forwardPrefillAt, forwardPrefillWithPast, GemmaModel.ChunkContext and the matching FunctionGemmaContract/ExportHarness additions landed in #415 and #417 without a dump refresh; apiCheck was red on develop. Purely additive, no removed lines. Co-Authored-By: Claude Sonnet 5 --- .../functiongemma/api/functiongemma.api | 19 +++++++++++++++++++ llm-inference/gemma/api/jvm/gemma.api | 13 +++++++++++++ 2 files changed, 32 insertions(+) diff --git a/llm-inference/functiongemma/api/functiongemma.api b/llm-inference/functiongemma/api/functiongemma.api index 8f6fcd3b..ea0b5139 100644 --- a/llm-inference/functiongemma/api/functiongemma.api +++ b/llm-inference/functiongemma/api/functiongemma.api @@ -1,14 +1,23 @@ public final class sk/ainet/models/functiongemma/FunctionGemmaContract { public static final field CONTRACT_VERSION I + public static final field DEFAULT_CHUNK I public static final field FN_PREFILL Ljava/lang/String; + public static final field FN_PREFILL_AT Ljava/lang/String; + public static final field FN_PREFILL_WITH_PAST Ljava/lang/String; public static final field FN_REDECODE Ljava/lang/String; + public static final field FN_REDECODE_AT Ljava/lang/String; public static final field FN_WITH_PAST Ljava/lang/String; public static final field INSTANCE Lsk/ainet/models/functiongemma/FunctionGemmaContract; public static final field K_FIRST_IN_OUTPUT Z public static final field PARAMETER_SCOPE Ljava/lang/String; public final fun manifestJson (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/lang/String; public final fun prefillArgs ()Ljava/util/List; + public final fun prefillAtOutputs (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/util/List; public final fun prefillOutputs (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/util/List; + public final fun prefillWithPastArgs (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/util/List; + public final fun prefillWithPastOutputs (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/util/List; + public final fun qualified (Ljava/lang/String;)Ljava/lang/String; + public final fun selectArgs ()Ljava/util/List; public final fun withPastArgs (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/util/List; public final fun withPastOutputs (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/util/List; } @@ -25,7 +34,17 @@ public final class sk/ainet/models/functiongemma/FunctionGemmaExportHarness { public final fun exportPrefill (Ljava/lang/String;Ljava/lang/String;IFZ)Ljava/lang/String; public final fun exportPrefill (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;)Ljava/lang/String; public static synthetic fun exportPrefill$default (Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness;Ljava/lang/String;Ljava/lang/String;IFZILjava/lang/Object;)Ljava/lang/String; + public final fun exportPrefillAt (Ljava/lang/String;Ljava/lang/String;IFZ)Ljava/lang/String; + public final fun exportPrefillAt (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;)Ljava/lang/String; + public static synthetic fun exportPrefillAt$default (Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness;Ljava/lang/String;Ljava/lang/String;IFZILjava/lang/Object;)Ljava/lang/String; + public final fun exportPrefillWithPast (Ljava/lang/String;Ljava/lang/String;IFZ)Ljava/lang/String; + public final fun exportPrefillWithPast (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;I)Ljava/lang/String; + public static synthetic fun exportPrefillWithPast$default (Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness;Ljava/lang/String;Ljava/lang/String;IFZILjava/lang/Object;)Ljava/lang/String; + public static synthetic fun exportPrefillWithPast$default (Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness;Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;IILjava/lang/Object;)Ljava/lang/String; public final fun exportRedecode (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;)Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness$RedecodeResult; + public final fun exportRedecodeAt (Ljava/lang/String;Ljava/lang/String;IFZ)Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness$RedecodeResult; + public final fun exportRedecodeAt (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;)Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness$RedecodeResult; + public static synthetic fun exportRedecodeAt$default (Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness;Ljava/lang/String;Ljava/lang/String;IFZILjava/lang/Object;)Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness$RedecodeResult; public final fun exportWithPast (Ljava/lang/String;Ljava/lang/String;IZFZ)Ljava/lang/String; public final fun exportWithPast (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;)Ljava/lang/String; public static synthetic fun exportWithPast$default (Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness;Ljava/lang/String;Ljava/lang/String;IZFZILjava/lang/Object;)Ljava/lang/String; diff --git a/llm-inference/gemma/api/jvm/gemma.api b/llm-inference/gemma/api/jvm/gemma.api index 13d030c0..4ba0ebe4 100644 --- a/llm-inference/gemma/api/jvm/gemma.api +++ b/llm-inference/gemma/api/jvm/gemma.api @@ -49,7 +49,10 @@ public final class sk/ainet/models/gemma/GemmaModel : sk/ainet/lang/nn/Module { public fun (Lsk/ainet/lang/nn/layers/EmbeddingAdapter;Lsk/ainet/models/gemma/PerLayerEmbedding;Ljava/util/List;Lsk/ainet/lang/nn/normalization/RMSNormalization;Lsk/ainet/lang/nn/transformer/VoidDense;Lkotlin/reflect/KClass;FFLjava/lang/String;)V public synthetic fun (Lsk/ainet/lang/nn/layers/EmbeddingAdapter;Lsk/ainet/models/gemma/PerLayerEmbedding;Ljava/util/List;Lsk/ainet/lang/nn/normalization/RMSNormalization;Lsk/ainet/lang/nn/transformer/VoidDense;Lkotlin/reflect/KClass;FFLjava/lang/String;ILkotlin/jvm/internal/DefaultConstructorMarker;)V public final fun buildRopeCosSin (ILsk/ainet/context/ExecutionContext;)Lsk/ainet/models/gemma/GemmaModel$RopeCosSin; + public final fun forwardAt (Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/context/ExecutionContext;)Lsk/ainet/lang/tensor/Tensor; public final fun forwardPrefill (Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/context/ExecutionContext;)Lsk/ainet/models/gemma/GemmaModel$GemmaPrefillOutput; + public final fun forwardPrefillAt (Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/context/ExecutionContext;)Lsk/ainet/models/gemma/GemmaModel$GemmaPrefillOutput; + public final fun forwardPrefillWithPast (Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/models/gemma/GemmaModel$ChunkContext;Lsk/ainet/lang/tensor/Tensor;Ljava/util/List;Ljava/util/List;Lsk/ainet/context/ExecutionContext;)Lsk/ainet/models/gemma/GemmaModel$GemmaWithPastOutput; public final fun forwardWithPast (Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/models/gemma/GemmaModel$RopeCosSin;Ljava/util/List;Ljava/util/List;Lsk/ainet/context/ExecutionContext;)Lsk/ainet/models/gemma/GemmaModel$GemmaWithPastOutput; public final fun getBlocks ()Ljava/util/List; public final fun getDtype ()Lkotlin/reflect/KClass; @@ -63,6 +66,16 @@ public final class sk/ainet/models/gemma/GemmaModel : sk/ainet/lang/nn/Module { public final fun getTokenEmbedding ()Lsk/ainet/lang/nn/layers/EmbeddingAdapter; } +public final class sk/ainet/models/gemma/GemmaModel$ChunkContext { + public fun (Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;)V + public final fun getCosGlobal ()Lsk/ainet/lang/tensor/Tensor; + public final fun getCosSliding ()Lsk/ainet/lang/tensor/Tensor; + public final fun getMaskGlobal ()Lsk/ainet/lang/tensor/Tensor; + public final fun getMaskSliding ()Lsk/ainet/lang/tensor/Tensor; + public final fun getSinGlobal ()Lsk/ainet/lang/tensor/Tensor; + public final fun getSinSliding ()Lsk/ainet/lang/tensor/Tensor; +} + public final class sk/ainet/models/gemma/GemmaModel$Companion { public final fun findHookIn (Ljava/util/List;)Lsk/ainet/models/gemma/PerLayerInputBlockHook; }