From 2f25f9ef7ef8b42969e88f5fd9e8de021962390e Mon Sep 17 00:00:00 2001 From: Saksham Gupta Date: Fri, 8 Dec 2023 00:24:11 +0530 Subject: [PATCH] nug fix --- sytorch/include/sytorch/tensor.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sytorch/include/sytorch/tensor.h b/sytorch/include/sytorch/tensor.h index 87dfd024..c07c2baf 100644 --- 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 d; + data[new_idx] = type_cast(d); #else data[new_idx] = type_cast(d * (1LL << scale)); #endif