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