wip

Author: czgdp1807

src/libasr/codegen/llvm_utils.cpp

@@ -85,6 +85,24 @@ namespace LFortran {
             };
             return builder.CreateCall(fn, args);
         }
+
+        llvm::Value* lfortran_free(llvm::LLVMContext &context, llvm::Module &module,
+                                   llvm::IRBuilder<> &builder, llvm::Value* ptr) {
+            std::string func_name = "_lfortran_free";
+            llvm::Function *fn = module.getFunction(func_name);
+            if (!fn) {
+                llvm::FunctionType *function_type = llvm::FunctionType::get(
+                        llvm::Type::getVoidTy(context), {
+                            llvm::Type::getInt8PtrTy(context)
+                        }, true);
+                fn = llvm::Function::Create(function_type,
+                        llvm::Function::ExternalLinkage, func_name, module);
+            }
+            std::vector<llvm::Value*> args = {
+                builder.CreateBitCast(ptr, llvm::Type::getInt8PtrTy(context)),
+            };
+            return builder.CreateCall(fn, args);
+        }
     } // namespace LLVM
 
     LLVMUtils::LLVMUtils(llvm::LLVMContext& context,
@@ -429,6 +447,9 @@ namespace LFortran {
         // integration_tests/test_list_07.py.
         if( LLVM::is_llvm_struct(element_type) ) {
             builder->CreateStore(copy_data, get_pointer_to_list_data(dest));
+            // TODO: Should be created outside the user loop and not here.
+            // LLVMList should treat them as data members and create them
+            // only if they are NULL
             llvm::AllocaInst *pos_ptr = builder->CreateAlloca(llvm::Type::getInt32Ty(context),
                                                               nullptr);
             LLVM::CreateStore(*builder, llvm::ConstantInt::get(llvm::Type::getInt32Ty(context),
@@ -708,6 +729,9 @@ namespace LFortran {
         new_key_mask = builder->CreateBitCast(new_key_mask, llvm::Type::getInt1PtrTy(context));
 
         llvm::Value* current_capacity = LLVM::CreateLoad(*builder, get_pointer_to_capacity(dict));
+        // TODO: Should be created outside the user loop and not here.
+        // LLVMDict should treat them as data members and create them
+        // only if they are NULL
         llvm::Value* idx_ptr = builder->CreateAlloca(llvm::Type::getInt32Ty(context), nullptr);
         LLVM::CreateStore(*builder, llvm::ConstantInt::get(llvm::Type::getInt32Ty(context),
             llvm::APInt(32, 0)), idx_ptr);
@@ -769,6 +793,9 @@ namespace LFortran {
         reset_iterators();
 
         // TODO: Free key_list, value_list and key_mask
+        llvm_utils->list_api->free_data(key_list, *module);
+        llvm_utils->list_api->free_data(value_list, *module);
+        LLVM::lfortran_free(context, *module, *builder, key_mask);
         LLVM::CreateStore(*builder, LLVM::CreateLoad(*builder, new_key_list), key_list);
         LLVM::CreateStore(*builder, LLVM::CreateLoad(*builder, new_value_list), value_list);
         LLVM::CreateStore(*builder, new_key_mask, get_pointer_to_keymask(dict));
@@ -925,10 +952,16 @@ namespace LFortran {
          * list[pos] = item;
          */
 
+        // TODO: Should be created outside the user loop and not here.
+        // LLVMList should treat them as data members and create them
+        // only if they are NULL
         llvm::AllocaInst *tmp_ptr = builder->CreateAlloca(el_type, nullptr);
         LLVM::CreateStore(*builder, read_item(list, pos, false), tmp_ptr);
         llvm::Value* tmp = nullptr;
 
+        // TODO: Should be created outside the user loop and not here.
+        // LLVMList should treat them as data members and create them
+        // only if they are NULL
         llvm::AllocaInst *pos_ptr = builder->CreateAlloca(
                                     llvm::Type::getInt32Ty(context), nullptr);
         LLVM::CreateStore(*builder, pos, pos_ptr);
@@ -975,6 +1008,9 @@ namespace LFortran {
         llvm::Type* pos_type = llvm::Type::getInt32Ty(context);
         llvm::Value* current_end_point = LLVM::CreateLoad(*builder,
                                         get_pointer_to_current_end_point(list));
+        // TODO: Should be created outside the user loop and not here.
+        // LLVMList should treat them as data members and create them
+        // only if they are NULL
         llvm::AllocaInst *i = builder->CreateAlloca(pos_type, nullptr);
         LLVM::CreateStore(*builder, llvm::ConstantInt::get(
                                     context, llvm::APInt(32, 0)), i);
@@ -1059,6 +1095,9 @@ namespace LFortran {
         llvm::Type* pos_type = llvm::Type::getInt32Ty(context);
         llvm::Value* current_end_point = LLVM::CreateLoad(*builder,
                                         get_pointer_to_current_end_point(list));
+        // TODO: Should be created outside the user loop and not here.
+        // LLVMList should treat them as data members and create them
+        // only if they are NULL
         llvm::AllocaInst *item_pos = builder->CreateAlloca(pos_type, nullptr);
         llvm::Value* tmp = LLVMList::find_item_position(list, item, item_type, module);
         LLVM::CreateStore(*builder, tmp, item_pos);
@@ -1114,6 +1153,11 @@ namespace LFortran {
         LLVM::CreateStore(*builder, zero, end_point_ptr);
     }
 
+    void LLVMList::free_data(llvm::Value* list, llvm::Module& module) {
+        llvm::Value* data = LLVM::CreateLoad(*builder, get_pointer_to_list_data(list));
+        LLVM::lfortran_free(context, module, *builder, data);
+    }
+
 
     LLVMTuple::LLVMTuple(llvm::LLVMContext& context_,
                          LLVMUtils* llvm_utils_,

src/libasr/codegen/llvm_utils.h

@@ -51,6 +51,8 @@ namespace LFortran {
                 llvm::IRBuilder<> &builder, llvm::Value* ptr, llvm::Value* arg_size);
         llvm::Value* lfortran_calloc(llvm::LLVMContext &context, llvm::Module &module,
                 llvm::IRBuilder<> &builder, llvm::Value* count, llvm::Value* type_size);
+        llvm::Value* lfortran_free(llvm::LLVMContext &context, llvm::Module &module,
+                llvm::IRBuilder<> &builder, llvm::Value* ptr);
         static inline bool is_llvm_struct(ASR::ttype_t* asr_type) {
             return ASR::is_a<ASR::Tuple_t>(*asr_type) ||
                    ASR::is_a<ASR::List_t>(*asr_type) ||
@@ -176,6 +178,8 @@ namespace LFortran {
             llvm::Value* find_item_position(llvm::Value* list,
                 llvm::Value* item, ASR::ttype_t* item_type,
                 llvm::Module& module);
+
+            void free_data(llvm::Value* list, llvm::Module& module);
     };
 
     class LLVMTuple {