From aad59d2055a8e428d48f725fb8c6502863eec00e Mon Sep 17 00:00:00 2001
From: lwj <fionser@gmail.com>
Date: Thu, 19 Sep 2024 12:53:53 +0800
Subject: [PATCH] [CHEETAH] fix eltype for mulaa (#857)

# Pull Request

## What problem does this PR solve?

Issue Number: Fixed #

## Possible side effects?

- Performance:

- Backward compatibility:
---
 libspu/mpc/cheetah/arithmetic.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/libspu/mpc/cheetah/arithmetic.cc b/libspu/mpc/cheetah/arithmetic.cc
index 1e2b32b9..75fbcc91 100644
--- a/libspu/mpc/cheetah/arithmetic.cc
+++ b/libspu/mpc/cheetah/arithmetic.cc
@@ -354,7 +354,7 @@ NdArrayRef MulAA::mulDirectly(KernelEvalContext* ctx, const NdArrayRef& x,
                          fy.slice({0}, {nhalf}, {1}), /*evaluato*/ rank != 0);
   out_slices[1] = task.get();
 
-  NdArrayRef out(out_slices[0].eltype(), x.shape());
+  NdArrayRef out(x.eltype(), x.shape());
   int64_t offset = 0;
   for (auto& out_slice : out_slices) {
     std::memcpy(out.data<std::byte>() + offset, out_slice.data(),