From 7ef7a2e914ba0cbc7e0da39a8f3a1413481bebe7 Mon Sep 17 00:00:00 2001 From: intellinjun Date: Thu, 29 Feb 2024 17:50:08 +0800 Subject: [PATCH] fix compile error Signed-off-by: intellinjun --- neural_speed/core/ne_layers.c | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/neural_speed/core/ne_layers.c b/neural_speed/core/ne_layers.c index 65b524891..16a1053d9 100644 --- a/neural_speed/core/ne_layers.c +++ b/neural_speed/core/ne_layers.c @@ -432,6 +432,7 @@ static const char* NE_OP_LABEL[NE_OP_COUNT] = { "FFN_SILU", "FFN_GeLU", "FFN_ADD_GeLU", + "FFN_ID_SILU", "FLASH_ATTN", "FLASH_ATTN_KV_UPDATE", "FLASH_FF", @@ -445,7 +446,7 @@ static const char* NE_OP_LABEL[NE_OP_COUNT] = { "DEBUG", }; -static_assert(NE_OP_COUNT == 66, "NE_OP_COUNT != 66"); +static_assert(NE_OP_COUNT == 67, "NE_OP_COUNT != 67"); static const char* NE_OP_SYMBOL[NE_OP_COUNT] = { "none", @@ -503,6 +504,7 @@ static const char* NE_OP_SYMBOL[NE_OP_COUNT] = { "QKV(x)", "ffn_silu(x)", + "ffn_id_silu(x)", "ffn_gelu(x)", "ffn_gelu_with_bias(x)", "flash_attn(x)",