[CHEETAH] fix eltype for mulaa

secretflow · Sep 19, 2024 · 2bfe1d8 · 2bfe1d8
1 parent f4533e2
commit 2bfe1d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(),