[onert] Add type validation check for RoPE

- check data type of sin/cos input ONE-DCO-1.0-Signed-off-by: youngsik kim <ys44.kim@samsung.com>
Samsung · Oct 2, 2024 · ad223c2 · ad223c2
1 parent c43dfa6
commit ad223c2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/runtime/onert/core/src/ir/OperationValidator.cc b/runtime/onert/core/src/ir/OperationValidator.cc
@@ -459,6 +459,8 @@ void OperationValidator::visit(const operation::RoPE &node)
   const auto output_index{node.getOutputs().at(operation::RoPE::Output::OUTPUT)};
 
   OP_REQUIRES(isValidType(input_index, DataType::FLOAT32));
+  OP_REQUIRES(isValidType(sin_index, DataType::FLOAT32));
+  OP_REQUIRES(isValidType(cos_index, DataType::FLOAT32));
   OP_REQUIRES(isSameType(input_index, output_index));
 }