File tree Expand file tree Collapse file tree 1 file changed +5
-0
lines changed
src/plugins/intel_cpu/src/nodes/kernels/scaled_attn Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Original file line number Diff line number Diff line change 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 ());
You can’t perform that action at this time.
0 commit comments