Merge pull request #1165 from IntelPython/fix/range_types_name

Improve __str__ for range_types
IntelPython · Oct 12, 2023 · 1dda097 · 1dda097
2 parents b9f4542 + 5b36457
commit 1dda097
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/numba_dpex/core/types/range_types.py b/numba_dpex/core/types/range_types.py
@@ -22,7 +22,7 @@ def __init__(self, ndim: int):
             raise errors.TypingError(
                 "RangeType can only have 1,2, or 3 dimensions"
             )
-        super(RangeType, self).__init__(name="Range")
+        super(RangeType, self).__init__(name="Range<" + str(ndim) + ">")
 
     @property
     def ndim(self):
@@ -44,7 +44,7 @@ def __init__(self, ndim: int):
             raise errors.TypingError(
                 "RangeType can only have 1,2, or 3 dimensions"
             )
-        super(NdRangeType, self).__init__(name="NdRange")
+        super(NdRangeType, self).__init__(name="NdRange<" + str(ndim) + ">")
 
     @property
     def ndim(self):