Merge pull request #2090 from Shaikh-Ubaid/cast_unsigned

Support cast UnsignedIntegerToUnsignedInteger

Author: ubaidsk

integration_tests/CMakeLists.txt

@@ -632,6 +632,7 @@ RUN(NAME vec_01              LABELS cpython llvm c NOFAST)
 RUN(NAME test_str_comparison LABELS cpython llvm c)
 RUN(NAME test_bit_length     LABELS cpython llvm c)
 RUN(NAME str_to_list_cast    LABELS cpython llvm c)
+RUN(NAME cast_01             LABELS cpython llvm c)
 RUN(NAME test_sys_01         LABELS cpython llvm c NOFAST)
 RUN(NAME intent_01           LABELS cpython llvm)
 

integration_tests/cast_01.py

@@ -0,0 +1,18 @@
+from lpython import i32, u8, u32, dataclass
+from numpy import empty, uint8
+
+@dataclass
+class LPBHV_small:
+    dim : i32 = 4
+    a : u8[4] = empty(4, dtype=uint8)
+
+def main0():
+    lphv_small : LPBHV_small = LPBHV_small()
+    i: i32
+    for i in range(4):
+        lphv_small.a[i] = u8(10 + i)
+        elt: u32 = u32(lphv_small.a[i])
+        print(elt)
+        assert elt == u32(10 + i)
+
+main0()

src/libasr/ASR.asdl

@@ -429,6 +429,7 @@ cast_kind
     | IntegerToCharacter
     | LogicalToCharacter
     | UnsignedIntegerToInteger
+    | UnsignedIntegerToUnsignedInteger
     | UnsignedIntegerToReal
     | UnsignedIntegerToLogical
     | IntegerToUnsignedInteger

src/libasr/codegen/asr_to_c_cpp.h

@@ -1754,8 +1754,10 @@ PyMODINIT_FUNC PyInit_lpython_module_)" + fn_name + R"((void) {
                 // src = src;
                 break;
             }
-            case (ASR::cast_kindType::IntegerToInteger) : {
+            case (ASR::cast_kindType::IntegerToInteger) :
+            case (ASR::cast_kindType::UnsignedIntegerToUnsignedInteger) : {
                 // In C++, we do not need to cast int <-> long long explicitly:
+                // we also do not need to cast uint8_t <-> uint32_t explicitly:
                 // src = src;
                 break;
             }

src/libasr/codegen/asr_to_llvm.cpp

@@ -6876,7 +6876,8 @@ class ASRToLLVMVisitor : public ASR::BaseVisitor<ASRToLLVMVisitor>
                 }
                 break;
             }
-            case (ASR::cast_kindType::IntegerToInteger) : {
+            case (ASR::cast_kindType::IntegerToInteger) :
+            case (ASR::cast_kindType::UnsignedIntegerToUnsignedInteger) : {
                 int arg_kind = -1, dest_kind = -1;
                 extract_kinds(x, arg_kind, dest_kind);
                 if( arg_kind > 0 && dest_kind > 0 &&

src/lpython/semantics/python_ast_to_asr.cpp

@@ -72,7 +72,8 @@ namespace CastingUtil {
     const std::map<ASR::ttypeType, ASR::cast_kindType>& kind_rules = {
         {ASR::ttypeType::Complex, ASR::cast_kindType::ComplexToComplex},
         {ASR::ttypeType::Real, ASR::cast_kindType::RealToReal},
-        {ASR::ttypeType::Integer, ASR::cast_kindType::IntegerToInteger}
+        {ASR::ttypeType::Integer, ASR::cast_kindType::IntegerToInteger},
+        {ASR::ttypeType::UnsignedInteger, ASR::cast_kindType::UnsignedIntegerToUnsignedInteger}
     };
 
     int get_type_priority(ASR::ttypeType type) {