From 6920cc996c221f1c4844efb15d427c416c4f8fae Mon Sep 17 00:00:00 2001
From: ysh329 <ysh329@users.noreply.github.com>
Date: Wed, 5 Aug 2020 06:18:56 -0500
Subject: [PATCH] [ARM] Support arm cpu cast from int32 to int64 (#4053)
 (#4055)

* add int32->int64 cast arm cpu kernel. test=develop
---
 lite/kernels/arm/cast_compute.cc | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/lite/kernels/arm/cast_compute.cc b/lite/kernels/arm/cast_compute.cc
index 919e9c603e..f17a83110b 100644
--- a/lite/kernels/arm/cast_compute.cc
+++ b/lite/kernels/arm/cast_compute.cc
@@ -72,6 +72,12 @@ void CastCompute::Run() {
     const int64_t* x_data_end = x_data_begin + param.X->numel();
     float* out_data = param.Out->mutable_data<float>();
     std::transform(x_data_begin, x_data_end, out_data, TransOp<int64_t, float>);
+  } else if (param.in_dtype == 2 && param.out_dtype == 3) {  // INT32 -> INT64
+    const int32_t* x_data_begin = param.X->data<int32_t>();
+    const int32_t* x_data_end = x_data_begin + param.X->numel();
+    int64_t* out_data = param.Out->mutable_data<int64_t>();
+    std::transform(
+        x_data_begin, x_data_end, out_data, TransOp<int32_t, int64_t>);
   } else {
     LOG(FATAL) << "other has not been implemented transform with dtype"
                << param.in_dtype << " X, dtype" << param.out_dtype << " Out";
-- 
GitLab