[C] Use memcpy for initializing SIMDArray's

Thirumalai-Shaktivel · Agent-Hellboy · commit 324f583956b9 · 2024-03-05T21:40:14.000+05:30
diff --git a/src/libasr/asr_utils.cpp b/src/libasr/asr_utils.cpp
@@ -1521,7 +1521,12 @@ void make_ArrayBroadcast_t_util(Allocator& al, const Location& loc,
     if (ret_type == nullptr) {
         // TODO: Construct appropriate return type here
         // For now simply coping the type from expr1
-        ret_type = expr1_type;
+        if (ASRUtils::is_simd_array(expr1)) {
+            // TODO: Make this more general; do not check for SIMDArray
+            ret_type = ASRUtils::duplicate_type(al, expr1_type);
+        } else {
+            ret_type = expr1_type;
+        }
     }
     expr2 = ASRUtils::EXPR(ASR::make_ArrayBroadcast_t(al, loc, expr2, dest_shape, ret_type, value));
 
diff --git a/src/libasr/codegen/asr_to_c_cpp.h b/src/libasr/codegen/asr_to_c_cpp.h
@@ -1246,7 +1246,37 @@ PyMODINIT_FUNC PyInit_lpython_module_)" + fn_name + R"((void) {
         bool is_value_dict = ASR::is_a<ASR::Dict_t>(*m_value_type);
         bool alloc_return_var = false;
         std::string indent(indentation_level*indentation_spaces, ' ');
-        if (ASR::is_a<ASR::Var_t>(*x.m_target)) {
+        if (ASRUtils::is_simd_array(x.m_target)) {
+            this->visit_expr(*x.m_target);
+            target = src;
+            if (ASR::is_a<ASR::Var_t>(*x.m_value) ||
+                    ASR::is_a<ASR::ArraySection_t>(*x.m_value)) {
+                std::string arr_element_type = CUtils::get_c_type_from_ttype_t(
+                    ASRUtils::expr_type(x.m_value));
+                std::string size = std::to_string(ASRUtils::get_fixed_size_of_array(
+                    ASRUtils::expr_type(x.m_target)));
+                std::string value;
+                if (ASR::is_a<ASR::ArraySection_t>(*x.m_value)) {
+                    ASR::ArraySection_t *arr = ASR::down_cast<ASR::ArraySection_t>(x.m_value);
+                    this->visit_expr(*arr->m_v);
+                    value = src;
+                    if(!ASR::is_a<ASR::ArrayBound_t>(*arr->m_args->m_left)) {
+                        this->visit_expr(*arr->m_args->m_left);
+                        int n_dims = ASRUtils::extract_n_dims_from_ttype(arr->m_type) - 1;
+                        value += "->data + (" + src + " - "+ value +"->dims["
+                            + std::to_string(n_dims) +"].lower_bound)";
+                    } else {
+                        value += "->data";
+                    }
+                } else if (ASR::is_a<ASR::Var_t>(*x.m_value)) {
+                    this->visit_expr(*x.m_value);
+                    value = src + "->data";
+                }
+                src = indent + "memcpy(&"+ target +", "+ value +", sizeof("
+                    + arr_element_type + ") * "+ size +");\n";
+                return;
+            }
+        } else if (ASR::is_a<ASR::Var_t>(*x.m_target)) {
             ASR::Var_t* x_m_target = ASR::down_cast<ASR::Var_t>(x.m_target);
             visit_Var(*x_m_target);
             target = src;
