Skip to content

Commit e7f467d

Browse files
committed
Dereference after null check
1 parent 6d052b6 commit e7f467d

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/plugins/intel_cpu/src/nodes/kernels/scaled_attn/softmax_kernel.hpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
#include <limits>
1212
#include <type_traits>
1313

14+
#include "openvino/core/except.hpp"
1415
#include "openvino/core/type/bfloat16.hpp"
1516
#include "openvino/core/type/element_type.hpp"
1617
#include "openvino/core/type/float16.hpp"
@@ -192,6 +193,10 @@ inline void scale_add2_reduce_max(float* a,
192193
size_t size,
193194
float alibi_slope,
194195
float& max) {
196+
OPENVINO_ASSERT(ov::intel_cpu::implication(has_alibi, alibi_lookup), "CPU: alibi_lookup should not be nullptr.");
197+
OPENVINO_ASSERT(ov::intel_cpu::implication(has_attn_mask, attn_mask), "CPU: attn_mask should not be nullptr.");
198+
OPENVINO_ASSERT(ov::intel_cpu::implication(has_causal_mask, causal_mask),
199+
"CPU: causal_mask should not be nullptr.");
195200
size_t i = 0;
196201
#if defined(HAVE_AVX512F)
197202
auto v_max0 = _mm512_set1_ps(std::numeric_limits<float>::lowest());

0 commit comments

Comments
 (0)