From bb9b62b169f975e2a570780e80c89dfde0a302f1 Mon Sep 17 00:00:00 2001
From: YdrMaster <ydrml@hotmail.com>
Date: Wed, 22 Feb 2023 11:05:54 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E6=94=B9=E6=AD=A3=E7=B1=BB=E5=9E=8B?=
 =?UTF-8?q?=E8=BD=AC=E6=8D=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: YdrMaster <ydrml@hotmail.com>
---
 src/ffi/ffi_infinitensor.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/ffi/ffi_infinitensor.cc b/src/ffi/ffi_infinitensor.cc
index f979b03c..c2d35051 100644
--- a/src/ffi/ffi_infinitensor.cc
+++ b/src/ffi/ffi_infinitensor.cc
@@ -95,17 +95,17 @@ static int tensor_dtype(Tensor t) {
 
 static int concat_axis_of(Operator op) {
     IT_ASSERT(op->getOpType() == OpType::Concat);
-    return reinterpret_cast<const ConcatObj *>(op.get())->getDim();
+    return dynamic_cast<const ConcatObj *>(op.get())->getDim();
 }
 
 static int gather_axis_of(Operator op) {
     IT_ASSERT(op->getOpType() == OpType::Gather);
-    return reinterpret_cast<const GatherObj *>(op.get())->getAxis();
+    return dynamic_cast<const GatherObj *>(op.get())->getAxis();
 }
 
 static Shape reshape_shape_of(Operator op) {
     IT_ASSERT(op->getOpType() == OpType::Reshape);
-    return reinterpret_cast<const ReshapeObj *>(op.get())->getShape();
+    return dynamic_cast<const ReshapeObj *>(op.get())->getShape();
 }
 
 void init_graph_builder(py::module &m) {