Small refactor with BlockExt (#939)

* extend blockext

* arugment -> arg

* extui

* extsi

* shrui ext

* shrui

* shli

* cmpi

* fix

* mul addi

* muli

* addi

* check

* trunci

* trunci

src/compiler.rs

@@ -793,7 +793,7 @@ fn compile_func(
                         block.append_operation(llvm::store(
                             context,
                             ptr,
-                            pre_entry_block.argument(0)?.into(),
+                            pre_entry_block.arg(0)?,
                             location,
                             LoadStoreOptions::new().align(Some(IntegerAttribute::new(
                                 IntegerType::new(context, 64).into(),

src/libfuncs/bitwise.rs

@@ -25,11 +25,10 @@ pub fn build<'ctx, 'this>(
     _metadata: &mut MetadataStorage,
     _info: &SignatureOnlyConcreteLibfunc,
 ) -> Result<()> {
-    let bitwise =
-        super::increment_builtin_counter(context, entry, location, entry.argument(0)?.into())?;
+    let bitwise = super::increment_builtin_counter(context, entry, location, entry.arg(0)?)?;
 
-    let lhs = entry.argument(1)?.into();
-    let rhs = entry.argument(2)?.into();
+    let lhs = entry.arg(1)?;
+    let rhs = entry.arg(2)?;
 
     let logical_and = entry.append_op_result(arith::andi(lhs, rhs, location))?;
     let logical_xor = entry.append_op_result(arith::xori(lhs, rhs, location))?;

src/libfuncs/bool.rs

@@ -100,8 +100,8 @@ fn build_bool_binary<'ctx, 'this>(
         .trailing_zeros();
     let tag_ty = IntegerType::new(context, tag_bits).into();
 
-    let lhs = entry.argument(0)?.into();
-    let rhs = entry.argument(1)?.into();
+    let lhs = entry.arg(0)?;
+    let rhs = entry.arg(1)?;
 
     let lhs_tag = entry.extract_value(context, location, lhs, tag_ty, 0)?;
 
@@ -149,7 +149,7 @@ pub fn build_bool_not<'ctx, 'this>(
         .trailing_zeros();
     let tag_ty = IntegerType::new(context, tag_bits).into();
 
-    let value = entry.argument(0)?.into();
+    let value = entry.arg(0)?;
     let tag_value = entry.extract_value(context, location, value, tag_ty, 0)?;
 
     let const_1 = entry.const_int_from_type(context, location, 1, tag_ty)?;
@@ -199,10 +199,10 @@ pub fn build_bool_to_felt252<'ctx, 'this>(
         .trailing_zeros();
     let tag_ty = IntegerType::new(context, tag_bits).into();
 
-    let value = entry.argument(0)?.into();
+    let value = entry.arg(0)?;
     let tag_value = entry.extract_value(context, location, value, tag_ty, 0)?;
 
-    let result = entry.append_op_result(arith::extui(tag_value, felt252_ty, location))?;
+    let result = entry.extui(tag_value, felt252_ty, location)?;
 
     entry.append_operation(helper.br(0, &[result], location));
     Ok(())