nug fix

mpc-msri · Dec 7, 2023 · 2f25f9e · 2f25f9e
1 parent fc2f88d
commit 2f25f9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sytorch/include/sytorch/tensor.h b/sytorch/include/sytorch/tensor.h
@@ -235,7 +235,7 @@ class Tensor {
             u64 curr_rest = i % rest_size;
             u64 new_idx = curr_batch * (num_channel * rest_size) + curr_rest * num_channel + curr_channel;
 #ifdef Do_Masking
-            data[new_idx] = type_cast<T> d;
+            data[new_idx] = type_cast<T>(d);
 #else
             data[new_idx] = type_cast<T>(d * (1LL << scale));
 #endif