From 5c05eca42fa737a5b831b728afed8e54c6a689e2 Mon Sep 17 00:00:00 2001 From: Biruk Mammo Date: Tue, 26 May 2026 10:40:50 -0700 Subject: [PATCH] Minor fix to include what is used in flash_attention.h. PiperOrigin-RevId: 921554497 --- gemma/flash_attention.h | 1 + 1 file changed, 1 insertion(+) diff --git a/gemma/flash_attention.h b/gemma/flash_attention.h index 47618c48..7e83a823 100644 --- a/gemma/flash_attention.h +++ b/gemma/flash_attention.h @@ -22,6 +22,7 @@ #include +#include "gemma/activations.h" #include "gemma/configs.h" #include "gemma/flash_structs.h" #include "gemma/kv_cache.h"