Prevent overflow by cast in KernelGenerator of acl_cl (#9303)

This commit prevents overflow by cast in KernelGenerator of acl_cl.

Signed-off-by: jiseob.jang <jiseob.jang@samsung.com>

diff --git a/runtime/neurun/backend/acl_cl/KernelGenerator.cc b/runtime/neurun/backend/acl_cl/KernelGenerator.cc
index dae98fe..6a47033 100644 (file)
--- a/runtime/neurun/backend/acl_cl/KernelGenerator.cc
+++ b/runtime/neurun/backend/acl_cl/KernelGenerator.cc
@@ -462,8 +462,8 @@ void KernelGenerator::visit(const model::operation::FullyConnected &node)
     const auto feature_size = _ctx.at(input_index).shape().num_elements();
 
     UNUSED_RELEASE(feature_size);
-    assert((batch_size * input_size) >= 0);
-    assert(feature_size == static_cast<uint64_t>(batch_size * input_size));
+    assert(batch_size >= 0 && input_size >= 0);
+    assert(feature_size == static_cast<uint64_t>(batch_size) * static_cast<uint64_t>(input_size));
 
     // for reshaping
     needs_reshape = true;