diff --git a/src/ops_infer/self_attention/self_attention_encoder_fusion_ops_runner.h b/src/ops_infer/self_attention/self_attention_encoder_fusion_ops_runner.h index 68fb16defe045c9e052b3cc632f388c5407b4328..0964eecc12c2fc2a0f57b16517c415437700032f 100644 --- a/src/ops_infer/self_attention/self_attention_encoder_fusion_ops_runner.h +++ b/src/ops_infer/self_attention/self_attention_encoder_fusion_ops_runner.h @@ -11,6 +11,7 @@ #define SELF_ATTENTION_ENCODER_FUSION_OPS_RUNNER_H + #include #include "atb/runner/ops_runner.h" #include "atb/infer_op_params.h"