From 2c0069fafb53ccb7a45a6815025dfcbd2882a36e Mon Sep 17 00:00:00 2001 From: Bin Cheng Date: Sat, 20 Jun 2020 15:42:12 +0800 Subject: [PATCH] Record and restore postorder information in breaking alias sccs. gcc/ PR tree-optimization/95638 * tree-loop-distribution.c (pg_edge_callback_data): New field. (loop_distribution::break_alias_scc_partitions): Record and restore postorder information. Fix memory leak. gcc/testsuite/ PR tree-optimization/95638 * g++.dg/tree-ssa/pr95638.C: New test. --- gcc/testsuite/g++.dg/tree-ssa/pr95638.C | 150 ++++++++++++++++++++++++ gcc/tree-loop-distribution.c | 23 +++- 2 files changed, 167 insertions(+), 6 deletions(-) create mode 100644 gcc/testsuite/g++.dg/tree-ssa/pr95638.C diff --git a/gcc/testsuite/g++.dg/tree-ssa/pr95638.C b/gcc/testsuite/g++.dg/tree-ssa/pr95638.C new file mode 100644 index 00000000000..d1bea6dffaa --- /dev/null +++ b/gcc/testsuite/g++.dg/tree-ssa/pr95638.C @@ -0,0 +1,150 @@ +// PR tree-optimization/95638 +// { dg-do run } +// { dg-options "-O2 -std=c++14" } + +#include +#include +#include +#include +#include +#include + +template +class intrusive_ref_counter +{ +public: + intrusive_ref_counter() = default; + + intrusive_ref_counter(intrusive_ref_counter const&) : + _counter(0) + { + } + + friend void intrusive_ptr_add_ref(const intrusive_ref_counter* p) + { + ++p->_counter; + } + + friend void intrusive_ptr_release(const intrusive_ref_counter* p) + { + if (--p->_counter == 0) + { + delete static_cast(p); + } + } + +private: + mutable int _counter = 0; +}; + +template +class intrusive_ptr +{ +public: + intrusive_ptr() = default; + + intrusive_ptr(T* p): px(p) + { + if (px != 0) intrusive_ptr_add_ref(px); + } + + intrusive_ptr(intrusive_ptr const & rhs): px(rhs.px) + { + if (px != 0) intrusive_ptr_add_ref(px); + } + + ~intrusive_ptr() + { + if (px != 0) intrusive_ptr_release(px); + } + + intrusive_ptr(intrusive_ptr && rhs) : px(rhs.px) + { + rhs.px = 0; + } + + explicit operator bool() const + { + return px != 0; + } + +private: + T* px = nullptr; +}; + +template +class Storage +{ +public: + Storage() = default; + + Storage(Storage&& other) + { + for (int i = 0; i < other._size; ++i) + { + new (data() + i) T(std::move(other.data()[i])); + ++_size; + } + } + + ~Storage() + { + for (int i = 0; i < _size; ++i) + { + data()[i].~T(); + } + } + + void push_back(const T& value) + { + assert(_size < MaxSize); + + new (data() + _size) T(value); + ++_size; + } + + T& operator[](size_t idx) + { + assert(idx < _size); + return data()[idx]; + } + +private: + T* data() + { + return reinterpret_cast(&_data); + } + +private: + uint32_t _size = 0; + std::aligned_storage_t _data; +}; + +struct Item: intrusive_ref_counter +{ +}; + +using Collection = Storage>; + +struct Holder +{ + __attribute__ ((noinline)) Holder(Collection collection) : collection(std::move(collection)) {} + + int64_t dummy = 0; // this is needed to reproduce the issue. + Collection collection; +}; + +int main() +{ + Collection collection; + collection.push_back(intrusive_ptr(new Item())); + + Holder holder(std::move(collection)); + + auto item = holder.collection[0]; + + if (!item) + __builtin_abort (); + + return 0; +} diff --git a/gcc/tree-loop-distribution.c b/gcc/tree-loop-distribution.c index b122c3964a0..9bc94e56a95 100644 --- a/gcc/tree-loop-distribution.c +++ b/gcc/tree-loop-distribution.c @@ -2145,6 +2145,8 @@ struct pg_edge_callback_data bitmap sccs_to_merge; /* Array constains component information for all vertices. */ int *vertices_component; + /* Array constains postorder information for all vertices. */ + int *vertices_post; /* Vector to record all data dependence relations which are needed to break strong connected components by runtime alias checks. */ vec *alias_ddrs; @@ -2401,7 +2403,7 @@ loop_distribution::break_alias_scc_partitions (struct graph *rdg, vec *partitions, vec *alias_ddrs) { - int i, j, k, num_sccs, num_sccs_no_alias; + int i, j, k, num_sccs, num_sccs_no_alias = 0; /* Build partition dependence graph. */ graph *pg = build_partition_graph (rdg, partitions, false); @@ -2452,6 +2454,7 @@ loop_distribution::break_alias_scc_partitions (struct graph *rdg, cbdata.sccs_to_merge = sccs_to_merge; cbdata.alias_ddrs = alias_ddrs; cbdata.vertices_component = XNEWVEC (int, pg->n_vertices); + cbdata.vertices_post = XNEWVEC (int, pg->n_vertices); /* Record the component information which will be corrupted by next graph scc finding call. */ for (i = 0; i < pg->n_vertices; ++i) @@ -2460,6 +2463,11 @@ loop_distribution::break_alias_scc_partitions (struct graph *rdg, /* Collect data dependences for runtime alias checks to break SCCs. */ if (bitmap_count_bits (sccs_to_merge) != (unsigned) num_sccs) { + /* Record the postorder information which will be corrupted by next + graph SCC finding call. */ + for (i = 0; i < pg->n_vertices; ++i) + cbdata.vertices_post[i] = pg->vertices[i].post; + /* Run SCC finding algorithm again, with alias dependence edges skipped. This is to topologically sort partitions according to compilation time known dependence. Note the topological order @@ -2490,11 +2498,6 @@ loop_distribution::break_alias_scc_partitions (struct graph *rdg, if (cbdata.vertices_component[k] != i) continue; - /* Update to the minimal postordeer number of vertices in scc so - that merged partition is sorted correctly against others. */ - if (pg->vertices[j].post > pg->vertices[k].post) - pg->vertices[j].post = pg->vertices[k].post; - partition_merge_into (NULL, first, partition, FUSE_SAME_SCC); (*partitions)[k] = NULL; partition_free (partition); @@ -2505,6 +2508,14 @@ loop_distribution::break_alias_scc_partitions (struct graph *rdg, first->type = PTYPE_SEQUENTIAL; } } + /* Restore the postorder information if it's corrupted in finding SCC + with alias dependence edges skipped. */ + if (num_sccs_no_alias > 0) + for (i = 0; i < pg->n_vertices; ++i) + pg->vertices[i].post = cbdata.vertices_post[i]; + + free (cbdata.vertices_component); + free (cbdata.vertices_post); } sort_partitions_by_post_order (pg, partitions); -- 2.30.2