From 403d0b7a4e13e8e3bcaa1a0c57ab66ffe761c7af Mon Sep 17 00:00:00 2001 From: Jonathon Penix Date: Thu, 29 Sep 2022 13:10:19 -0700 Subject: [flang] Skip creating AggregateStores for common block associated aggregates Previously, AggregateStores were created for aggregates associated with common blocks. As a) AggregateStoreMap uses scope and offset information to search for aggregate stores and b) variables related to common blocks have their offsets set relative to the common block itself, if there were multiple equivalences and at least one involved variables defined in a common block there was an opportunity for the scope/offset pairs to match between distinct aggregate stores. As a result, entries in AggregateStoreMap could collide, resulting in incorrect stores being returned for a particular variable. To prevent these collisions, skip creating AggregateStores for aggregates which are associated with common blocks. This information was already unused as aggregates associated with common blocks are handled by instantiateCommon. Fixes https://github.com/llvm/llvm-project/issues/57749 Differential Revision: https://reviews.llvm.org/D134828 --- flang/lib/Lower/PFTBuilder.cpp | 7 +++++++ flang/test/Lower/equivalence-2.f90 | 31 +++++++++++++++++++++++++++++++ 2 files changed, 38 insertions(+) (limited to 'flang') diff --git a/flang/lib/Lower/PFTBuilder.cpp b/flang/lib/Lower/PFTBuilder.cpp index 1c93fd4..62ec5ad 100644 --- a/flang/lib/Lower/PFTBuilder.cpp +++ b/flang/lib/Lower/PFTBuilder.cpp @@ -1324,6 +1324,13 @@ struct SymbolDependenceDepth { const Fortran::semantics::Symbol *aggregateSym = nullptr; bool isGlobal = false; const semantics::Symbol &first = *aggregate.front(); + // Skip aggregates related to common blocks as they will be handled by + // instantiateCommon and the aggregate store information will not be used. + // Additionally, the AggregateStoreKeys for common block related aggregate + // stores can collide with non common block ones, potentially resulting in + // incorrect stores being used. + if (lower::definedInCommonBlock(first)) + continue; std::size_t start = first.offset(); std::size_t end = first.offset() + first.size(); const Fortran::semantics::Symbol *namingSym = nullptr; diff --git a/flang/test/Lower/equivalence-2.f90 b/flang/test/Lower/equivalence-2.f90 index 476a9a4..610e741 100644 --- a/flang/test/Lower/equivalence-2.f90 +++ b/flang/test/Lower/equivalence-2.f90 @@ -97,3 +97,34 @@ end ! CHECK: %[[xCast:.*]] = fir.convert %[[x]] : (!fir.ptr>) -> !fir.ref> ! CHECK: %[[iCast:.*]] = fir.convert %[[i]] : (!fir.ptr) -> !fir.ref ! CHECK: fir.call @_QPfoo2(%[[xCast]], %[[iCast]]) : (!fir.ref>, !fir.ref) -> () + + +! Check that cases where equivalenced local variables and common blocks will +! share the same offset use the correct stores +! CHECK-LABEL: @_QPeq_and_comm_same_offset() +subroutine eq_and_comm_same_offset + real common_arr1(133),common_arr2(133) + common /my_common_block/ common_arr1,common_arr2 + real arr1(133),arr2(133) + real arr3(133,133),arr4(133,133) + equivalence(arr1,common_arr1),(arr2,common_arr2) + equivalence(arr3,arr4) + + ! CHECK: %[[arr4Store:.*]] = fir.alloca !fir.array<70756xi8> {uniq_name = "_QFeq_and_comm_same_offsetEarr3"} + ! CHECK: %[[mcbAddr:.*]] = fir.address_of(@_QBmy_common_block) : !fir.ref> + ! CHECK: %[[mcbCast:.*]] = fir.convert %[[mcbAddr]] : (!fir.ref>) -> !fir.ref> + ! CHECK: %[[c0:.*]] = arith.constant 0 : index + ! CHECK: %[[mcbCoor:.*]] = fir.coordinate_of %[[mcbCast]], %[[c0]] : (!fir.ref>, index) -> !fir.ref + ! CHECK: %[[mcbCoorCast:.*]] = fir.convert %[[mcbCoor]] : (!fir.ref) -> !fir.ptr> + ! CHECK: %[[c1:.*]] = arith.constant 0 : index + ! CHECK: %[[arr4Addr:.*]] = fir.coordinate_of %[[arr4Store]], %[[c1]] : (!fir.ref>, index) -> !fir.ref + ! CHECK: %[[arr4Cast:.*]] = fir.convert %[[arr4Addr]] : (!fir.ref) -> !fir.ptr> + + arr1(1) = 1 + ! CHECK:%[[mcbFinalAddr:.*]] = fir.coordinate_of %[[mcbCoorCast]], %{{.*}} : (!fir.ptr>, i64) -> !fir.ref + ! CHECK:fir.store %{{.*}} to %[[mcbFinalAddr]] : !fir.ref + + arr4(1,1) = 2 + ! CHECK: %[[arr4FinalAddr:.*]] = fir.coordinate_of %[[arr4Cast]], %{{.*}}, %{{.*}} : (!fir.ptr>, i64, i64) -> !fir.ref + ! CHECK: fir.store %{{.*}} to %[[arr4FinalAddr]] : !fir.ref +end subroutine -- cgit v1.1