Skip to content

Commit 10b9d71

Browse files
authored
[NNAdapter] Support int64 datatype for eltwise_ops (#9859)
1 parent 1e04c14 commit 10b9d71

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

lite/backends/nnadapter/nnadapter/src/operation/elementwise.cc

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,18 @@ NNADAPTER_EXPORT int ExecuteElementwise(core::Operation* operation) {
141141
input1_shape,
142142
static_cast<math::FuseCode>(fuse_code),
143143
output_data);
144+
} else if (input0_type.precision == NNADAPTER_INT64 &&
145+
input1_type.precision == NNADAPTER_INT64) {
146+
const auto input0_data = reinterpret_cast<const int64_t*>(input0_buffer);
147+
const auto input1_data = reinterpret_cast<const int64_t*>(input1_buffer);
148+
auto output_data = reinterpret_cast<int64_t*>(output_buffer);
149+
status = math::elementwise<int64_t>(eltwise_type,
150+
input0_data,
151+
input0_shape,
152+
input1_data,
153+
input1_shape,
154+
static_cast<math::FuseCode>(fuse_code),
155+
output_data);
144156
} else if (input0_type.precision == NNADAPTER_QUANT_INT8_SYMM_PER_LAYER &&
145157
input1_type.precision == NNADAPTER_QUANT_INT8_SYMM_PER_LAYER) {
146158
const auto input0_data = reinterpret_cast<const int8_t*>(input0_buffer);

0 commit comments

Comments
 (0)