From 2bfe1d8d3bb0dbe1f23a399e3f997d4d4469d302 Mon Sep 17 00:00:00 2001
From: fionser <fionser@gmail.com>
Date: Thu, 19 Sep 2024 10:51:53 +0800
Subject: [PATCH] [CHEETAH] fix eltype for mulaa

---
 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(),