From 992858b687d9ff53b53239864114ff7ece22449d Mon Sep 17 00:00:00 2001 From: gingerBill Date: Wed, 5 Feb 2020 22:39:23 +0000 Subject: Add ReturnStmt --- src/types.cpp | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'src/types.cpp') diff --git a/src/types.cpp b/src/types.cpp index fde8cd5a5..2afba733c 100644 --- a/src/types.cpp +++ b/src/types.cpp @@ -1143,13 +1143,13 @@ bool is_type_simd_vector(Type *t) { } Type *base_array_type(Type *t) { - if (is_type_array(t)) { - t = base_type(t); - return t->Array.elem; - } - if (is_type_simd_vector(t)) { - t = base_type(t); - return t->SimdVector.elem; + Type *bt = base_type(t); + if (is_type_array(bt)) { + return bt->Array.elem; + } else if (is_type_enumerated_array(bt)) { + return bt->EnumeratedArray.elem; + } else if (is_type_simd_vector(bt)) { + return bt->SimdVector.elem; } return t; } -- cgit v1.2.3