From: Eelis <eelis@eelis.net>
To: gcc-patches@gcc.gnu.org
Cc: libstdc++@gcc.gnu.org
Subject: Re: [patch, libstdc++] std::shuffle: Generate two swap positions at a time if possible
Date: Fri, 02 Sep 2016 19:27:00 -0000 [thread overview]
Message-ID: <57C9CAAE.50202@eelis.net> (raw)
Message-ID: <20160902192700.v0SPL1L0Lmbcubiako4xXpU0eL1aDguNNHl36p9LU_4@z> (raw)
In-Reply-To: <57C9C304.90607@eelis.net>
[-- Attachment #1: Type: text/plain, Size: 1093 bytes --]
On 2016-09-02 20:20, Eelis van der Weegen wrote:
> On 2016-08-31 14:45, Jonathan Wakely wrote:
>> Is this significantly faster than just using
>> uniform_int_distribution<_IntType>{0, __bound - 1}(__g) so we don't
>> need to duplicate the logic? (And people maintaining the code won't
>> reconvince themselves it's correct every time they look at it :-)
>>
>> [..]
>>
>> Could we hoist this test out of the loop somehow?
>>
>> If we change the loop condition to be __i+1 < __last we don't need to
>> test it on every iteration, and then after the loop we can just do
>> the final swap if (__urange % 2).
>
> Reusing std::uniform_int_distribution seems just as fast, so I've removed __generate_random_index_below.
>
> I've hoisted the (__i + 1 == __last) check out of the loop (in a slightly different way), and it seems to shave off a couple more cycles, yay!
>
> Updated patch attached.
>
Please ignore that patch, I used __g()&1 but that's invalid (the new "UniformRandomBitGenerator" name is misleading).
Updated patch (which uses a proper distribution even for the [0,1] case) attached.
[-- Attachment #2: newer-double-step-shuffle.patch --]
[-- Type: text/x-patch, Size: 1965 bytes --]
Index: libstdc++-v3/include/bits/stl_algo.h
===================================================================
--- libstdc++-v3/include/bits/stl_algo.h (revision 239895)
+++ libstdc++-v3/include/bits/stl_algo.h (working copy)
@@ -3772,6 +3772,47 @@
typedef typename std::make_unsigned<_DistanceType>::type __ud_type;
typedef typename std::uniform_int_distribution<__ud_type> __distr_type;
typedef typename __distr_type::param_type __p_type;
+
+ typedef typename std::remove_reference<_UniformRandomNumberGenerator>::type _Gen;
+ typedef typename std::common_type<typename _Gen::result_type, __ud_type>::type __uc_type;
+
+ const __uc_type __urngrange = __g.max() - __g.min();
+ const __uc_type __urange = __uc_type(__last - __first);
+
+ if (__urngrange / __urange >= __urange)
+ // I.e. (__urngrange >= __urange * __urange) but without wrap issues.
+ {
+ _RandomAccessIterator __i = __first + 1;
+
+ // Since we know the range isn't empty, an even number of elements
+ // means an uneven number of elements /to swap/, in which case we
+ // do the first one up front:
+
+ if ((__urange % 2) == 0)
+ {
+ __distr_type __d{0, 1};
+ std::iter_swap(__i++, __first + __d(__g));
+ }
+
+ // Now we know that __last - __i is even, so we do the rest in pairs,
+ // using a single distribution invocation to produce swap positions
+ // for two successive elements at a time:
+
+ while (__i != __last)
+ {
+ const __uc_type __swap_range = __uc_type(__i - __first) + 1;
+ const __uc_type __comp_range = __swap_range * (__swap_range + 1);
+
+ std::uniform_int_distribution<__uc_type> __d{0, __comp_range - 1};
+ const __uc_type __pospos = __d(__g);
+
+ std::iter_swap(__i++, __first + (__pospos % __swap_range));
+ std::iter_swap(__i++, __first + (__pospos / __swap_range));
+ }
+
+ return;
+ }
+
__distr_type __d;
for (_RandomAccessIterator __i = __first + 1; __i != __last; ++__i)
next prev parent reply other threads:[~2016-09-02 18:53 UTC|newest]
Thread overview: 16+ messages / expand[flat|nested] mbox.gz Atom feed top
2016-05-01 14:18 Eelis
2016-05-01 14:30 ` Eelis
2016-05-03 12:39 ` Jonathan Wakely
2016-05-03 14:42 ` Eelis van der Weegen
2016-05-25 19:54 ` Eelis
2016-05-25 20:45 ` Eelis
2016-08-31 12:45 ` Jonathan Wakely
2016-09-01 15:14 ` Jonathan Wakely
2016-09-01 15:27 ` Marc Glisse
2016-09-01 15:35 ` Jonathan Wakely
2016-09-01 15:31 ` Eelis van der Weegen
2016-09-01 15:37 ` Jonathan Wakely
2016-09-02 18:20 ` Eelis van der Weegen
2016-09-02 18:53 ` Eelis [this message]
2016-09-02 19:27 ` Eelis
2016-10-14 19:42 ` Jonathan Wakely
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-all from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
--in-reply-to=57C9CAAE.50202@eelis.net \
--to=eelis@eelis.net \
--cc=gcc-patches@gcc.gnu.org \
--cc=libstdc++@gcc.gnu.org \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox;
as well as URLs for read-only IMAP folder(s) and NNTP newsgroup(s).