在 uint64 中对 38 位进行基数排序?

Posted

技术标签:

【中文标题】在 uint64 中对 38 位进行基数排序?【英文标题】:Radix sort on 38 bits in a uint64? 【发布时间】:2020-01-28 17:34:46 【问题描述】:

问题:是否有“Dial_A_Bit”基数排序可用于对数据类型中的位子集进行排序?

uint64 *Radix_Sort_Dial_A_Bit(uint64_t *a, int num_a, int sort_bits);

具体来说,是否可以对 64 位数据进行 38 位排序,并且其速度介于如下所示的 32/64 和 48/64 之间?

uint64 *Radix_Sort_ui64_38MSB(uint64_t *a, int num_a);

请注意,与对 uint64_t[] 中的所有 64 位进行排序相比,对 48 位和 32 位排序的研究如何验证了速度和正确性的提高。

似乎对数据包大小的子集进行排序的 Radix_Sort() 通常有用且高效,只对需要的内容进行排序。

在某些情况下,结果是针对每个 Pixel 计算并需要排序的。一个 uint64_t[] 用于保存计算结果和 XY 位置。

总共需要 26 位(X 为 13,Y 为 13,最大分辨率为 8192)来保存像素 XY 坐标,剩下 38 位用于数据排序。

可以使用 Radix_Sort_Uint64() 对整个 64 位包进行排序。

一种更快的方法是使用 Radix_Sort_Uint48()(见下文),因此在排序中不考虑最后 16 位。这对所有数据进行了正确排序,并对 13 个 X 坐标位中的 10 个进行了排序,这是不需要的。

由于性能几乎与排序的位成线性比例,因此在最佳情况下,排序中只考虑 38 个最高有效位。

即使是 40 位基数排序也比使用 48 位要好。我试图将工作的 48 位基数排序概括为在 40 位上运行,但它不能正确排序。

QSort_uint64_38_msb():

static inline int comp_uint64_38_msb(const void *a, const void *b)  
register int64_t ca, cb;
    ca = (*((uint64_t *)a)) >> 26;  // Send 26 LSBs to oblivion
    cb = (*((uint64_t *)b)) >> 26;  // Send 26 LSBs to oblivion
    return((ca > cb) - (ca < cb));  // Calcs to [+1, 0 -1]

如您所见,48 位排序比完整的 64 位排序要快得多。 32 位排序几乎是完整 64 位排序的两倍。而 qsort 远远落后:

  Time=  2.136 sec =  3.390%, RADIX_SORT_FFFF0000  , hits=4, 0.534 sec each
  Time=  2.944 sec =  4.672%, RADIX_SORT_FFFFFF00  , hits=4, 0.736 sec each
  Time=  4.691 sec =  7.444%, RADIX_SORT_64        , hits=5, 0.938 sec each
  Time= 25.209 sec = 40.010%, QSORT_UINT64_ARRAY   , hits=4, 6.302 sec each
  Time= 26.191 sec = 41.569%, QSORT_UINT64_38_ARRAY, hits=4, 6.548 sec each

从 64、48 和 32 位结果线性化到 38 位:

lsf 64  0.938   48  0.736    32  0.534   38   ->  0.6500

38 位 Radix_Sort 比 64 位排序快 35%,比 48 位排序快 17%。

即使是 40 位也会更快,5 个字节与每个 uint64 处理 6 个字节。

=========

目前最快,6 个 8 字节的 uint64[],概括自: 32 MSbits used to sort uint64

// #############################################################################
// From: http://ideone.com/JHI0d9
// RadixSort---  for 48 MSB of uint64
typedef union 
    struct 
        uint32_t c6[256];
        uint32_t c5[256];
        uint32_t c4[256];
        uint32_t c3[256];
        uint32_t c2[256];
        uint32_t c1[256];
    ;
    uint32_t counts[256 * 6];
  rscounts6_t;


// #############################################################################
// Patterned off of Radix_Sort_64 but looks only at the 48 MostSigBits
// 0XFFFF-FFFF-FFFF-0000  << Ignore the zeros, sort on 3 MostSigBytes
// Made for RGB48 stuffed into uint64 with 2 LeastSig bytes zero
// Get rid of the 7 and 8 level comps
uint64_t *radix_sort_48_msb(uint64_t *arrayA, uint32_t asize) 

register uint64_t *array=arrayA;  // Slam arg into Register!
register int ii;  // Loop control

    rscounts6_t counts;
    memset(&counts, 0, 256 * 6 * sizeof(uint32_t));
    uint64_t *cpy = (uint64_t *)malloc(asize * sizeof(uint64_t));
    uint32_t o6=0, o5=0, o4=0, o3=0, o2=0, o1=0;
    uint32_t t6, t5, t4, t3, t2, t1;
    register uint32_t x;
    // calculate counts
    for(x = 0; x < asize; x++) 
        t6 = (array[x] >> 16) & 0xff;
        t5 = (array[x] >> 24) & 0xff;
        t4 = (array[x] >> 32) & 0xff;
        t3 = (array[x] >> 40) & 0xff;
        t2 = (array[x] >> 48) & 0xff;
        t1 = (array[x] >> 56) & 0xff;
        counts.c6[t6]++;
        counts.c5[t5]++;
        counts.c4[t4]++;
        counts.c3[t3]++;
        counts.c2[t2]++;
        counts.c1[t1]++;
    
    // convert counts to offsets
    for(x = 0; x < 256; x++) 
        t6 = o6 + counts.c6[x];
        t5 = o5 + counts.c5[x];
        t4 = o4 + counts.c4[x];
        t3 = o3 + counts.c3[x];
        t2 = o2 + counts.c2[x];
        t1 = o1 + counts.c1[x];
        counts.c6[x] = o6;
        counts.c5[x] = o5;
        counts.c4[x] = o4;
        counts.c3[x] = o3;
        counts.c2[x] = o2;
        counts.c1[x] = o1;
        o6 = t6; 
        o5 = t5; 
        o4 = t4; 
        o3 = t3; 
        o2 = t2; 
        o1 = t1;
    
    // radix
    for(x = 0; x < asize; x++) 
        t6 = (array[x] >> 16) & 0xff;
        cpy[counts.c6[t6]] = array[x];
        counts.c6[t6]++;  
    for(x = 0; x < asize; x++) 
        t5 = (cpy[x] >> 24) & 0xff;
        array[counts.c5[t5]] = cpy[x];
        counts.c5[t5]++;  
    for(x = 0; x < asize; x++) 
        t4 = (array[x] >> 32) & 0xff;
        cpy[counts.c4[t4]] = array[x];
        counts.c4[t4]++;  
    for(x = 0; x < asize; x++) 
        t3 = (cpy[x] >> 40) & 0xff;
        array[counts.c3[t3]] = cpy[x];
        counts.c3[t3]++;  
    for(x = 0; x < asize; x++) 
        t2 = (array[x] >> 48) & 0xff;
        cpy[counts.c2[t2]] = array[x];
        counts.c2[t2]++;  
    for(x = 0; x < asize; x++) 
        t1 = (cpy[x] >> 56) & 0xff;
        array[counts.c1[t1]] = cpy[x];
        counts.c1[t1]++;  
    free(cpy);
    return array;
  // End radix_sort_48_msb().

===================================

再次感谢 Rcgldr 提供创新的编程建议! 我没有使用 10、10、9、9,而是使用带有 [4][10]

的快速 32 位模式

它可以工作,但它比 48 MSB 排序慢很多,40 MSB 为 737 毫秒,48 MSB 为 588 毫秒。 :(

也许我编码不好。

    Time=  6.108 sec = 33.668%, QSORT_UINT64_ARRAY   , hits=1
    Time=  3.060 sec = 16.866%, RADIX_SORT_UINT64_REG, hits=4, 0.765 sec each
    Time=  2.947 sec = 16.241%, RADIX_SORT_UINT64_40R, hits=4, 0.737 sec each < SLOW
    Time=  2.354 sec = 12.973%, RADIX_SORT_UINT64_48R, hits=4, 0.588 sec each
    Time=  1.542 sec =  8.498%, RADIX_SORT_UINT64_32R, hits=4, 0.385 sec each
    Time=  0.769 sec =  4.236%, RADIX_SORT_64        , hits=1

测试:

创建一个随机的 uint64_t[36M] 主数组 使用 qsort 和已知良好的 Radix Sort 进行排序,radixSort() 以创建标准数组 比较 Qsort 和 radixSort() 结果的相同性。 使用 32、40、48 和 64 MSB 基数排序对主副本进行排序 在屏蔽掉忽略的 LSB 后,将每个测试排序与标准进行比较

代码如下:

    //=============================================================================
// From code submitted by rcgldr, Feb 8 2020
// Optimized to use Registers and to sort on 40 MSBs, ignoring 24 LSBs
void radix_sort_r64_40(uint64_t *pData, uint64_t *pTemp, size_t count,
    EV_TIME_STR *tsa)

    size_t mIndex[4][1024] =  0 ;  /* index matrix */
    size_t * pmIndex;                /* ptr to row of matrix */
    size_t i, j, m, n;
    uint64_t u;
    if(tsa)  time_event(E_RADIX_SORT_UINT64_40R, tsa, E_TIME_EVENT, 1, 0);  

    for (i = 0; i < count; i++)        /* generate histograms */
        u = pData[i];
        mIndex[3][(u >> 24) & 0x3ff]++;
        mIndex[2][(u >> 34) & 0x3ff]++;
        mIndex[1][(u >> 44) & 0x3ff]++;
        mIndex[0][(u >> 54) & 0x3ff]++;
    

    for (j = 0; j < 4; j++)            /* convert to indices */
        pmIndex = mIndex[j];
        n = 0;
        for (i = 0; i < 1024; i++)  
            m = pmIndex[i];
            pmIndex[i] = n;
            n += m;
        
    

    for (i = 0; i < count; i++)    /* radix sort */
        u = pData[i];
        pTemp[mIndex[3][(u >> 24) & 0x3ff]++] = u;
    
    for (i = 0; i < count; i++)  
        u = pTemp[i];
        pData[mIndex[2][(u >> 34) & 0x3ff]++] = u;
    
    for (i = 0; i < count; i++)  
        u = pData[i];
        pTemp[mIndex[1][(u >> 44) & 0x3ff]++] = u;
    
    for (i = 0; i < count; i++)  
        u = pTemp[i];
        pData[mIndex[0][(u >> 54) & 0x3ff]++] = u;
    
  // End Radix_Sort_R64_40().

这是 FAST 32 MSB 版本和克隆的 40 MSB 慢速版本之间的唯一差异。

Unique lines from "~/tmp/radix.sort.32.c":
  02) void radix_sort_r64_32(uint64_t *pData, uint64_t *pTemp, size_t count,
  05) size_t mIndex[4][256] =  0 ;      /* index matrix */
  09) if(tsa)  time_event(E_RADIX_SORT_UINT64_32R, tsa, E_TIME_EVENT, 1, 0);
  13) mIndex[3][(u >> 32) & 0xff]++;  // B4
  14) mIndex[2][(u >> 40) & 0xff]++;  // B5
  15) mIndex[1][(u >> 48) & 0xff]++;  // B6
  16) mIndex[0][(u >> 56) & 0xff]++;  // B7
  22) for (i = 0; i < 256; i++) 
  31) pTemp[mIndex[3][(u >> 32) & 0xff]++] = u;
  35) pData[mIndex[2][(u >> 40) & 0xff]++] = u;
  39) pTemp[mIndex[1][(u >> 48) & 0xff]++] = u;
  43) pData[mIndex[0][(u >> 56) & 0xff]++] = u;

Unique lines from "~/tmp/radix.sort.40.c":
  01) void radix_sort_r64_40(uint64_t *pData, uint64_t *pTemp, size_t count,
  04) size_t mIndex[4][1024] =  0 ;  /* index matrix */
  08) if(tsa)  time_event(E_RADIX_SORT_UINT64_40R, tsa, E_TIME_EVENT, 1, 0);
  12) mIndex[3][(u >> 24) & 0x3ff]++;
  13) mIndex[2][(u >> 34) & 0x3ff]++;
  14) mIndex[1][(u >> 44) & 0x3ff]++;
  15) mIndex[0][(u >> 54) & 0x3ff]++;
  21) for (i = 0; i < 1024; i++)  
  30) pTemp[mIndex[3][(u >> 24) & 0x3ff]++] = u;
  34) pData[mIndex[2][(u >> 34) & 0x3ff]++] = u;
  38) pTemp[mIndex[1][(u >> 44) & 0x3ff]++] = u;
  42) pData[mIndex[0][(u >> 54) & 0x3ff]++] = u;

【问题讨论】:

这是个问题吗? 一些不同(但不相关)的点:首先,在 C 中你不需要而且真的 shouldn't cast the result of malloc。同样要避免的是magic numbers(尤其是当sizeof 运算符可以得到“幻数”时,如sizeof counts)。而register 只是一个建议,如果编译器愿意,编译器根本不必费心(除了它必须遵循链接语义,比如没有获取地址等)。 @rcgldr - Skylake i7-6700K CPU @ 4.00GHz, cache_alignment : 64, AVX2, 64 GB DDR4 3.2GHz, CL16 (F4-3200C16-16GVK) 设置为 Interleaved 以实现更快的大块顺序。 4x32k inst/dat L1、4x256k L2、8M L3 @rcgldr - 您提到缓存一致性是 10 位 bin 与 8 位的性能下降的可能原因。写入是在总数组大小范围内从 A 点到 B 点完成的,所以它不依赖于数组大小并且独立于 bin 位吗?为了测试这一点,可以将阵列大小从 1M 更改为 32M,然后查看性能下降的地方。这可能会显示一个性能悬崖,指示缓存被淹没的位置。新的 AMD 3950x 具有 64 MB L3 和总共 8 MB L2!希望我有一个可以测试! @BrianP007 - 缓存一致性 - 与内核或处理器之间的共享内存有关,这里不是问题。我指的是 L1 缓存的大小,即每个内核 32KB(6700K 和 3770K 的缓存大小相同)。 [8][256] 64 位索引的矩阵占用 16KB 空间,留出空间 L1 缓存用于数组读取。 [4][1024] 64 位索引的矩阵占用 32KB 空间,在初始计数过程中从 size_t 切换到 uint32_t 会有所帮助。在基数排序期间,一次只使用一行,所以不是问题。 【参考方案1】:

可以使用 10 bit, 10 bit, 9 bit, 9 bit 计数 == 偏移在 4 遍中完成 38 msb 的基数排序。更改您的代码以使用大小:c1[1024]、c2[1024]、c3[512]、c4[512] 和计数初始化以使用移位和掩码 (...>>54)&0x3ff ,(.. .>>44)&0x3ff, (...>>35)&0x1ff,(...>>26)&0x1ff,并对其余代码进行类似的更改。


尝试使用此代码进行比较。更改您的代码或将计时器内容添加到此代码中:

void RadixSort(uint64_t *a, uint64_t *b, size_t count)

    uint32_t mIndex[4][1024] =  0 ;  /* index matrix */
    uint32_t * pmIndex;                /* ptr to row of matrix */
    uint32_t i, j, m, n;
    uint64_t u;

    for (i = 0; i < count; i++)        /* generate histograms */
        u = a[i];
        mIndex[3][(u >> 26) & 0x1ff]++;
        mIndex[2][(u >> 35) & 0x1ff]++;
        mIndex[1][(u >> 44) & 0x3ff]++;
        mIndex[0][(u >> 54) & 0x3ff]++;
    

    for (j = 0; j < 2; j++)            /* convert to indices */
        pmIndex = mIndex[j];
        n = 0;
        for (i = 0; i < 1024; i++)  
            m = pmIndex[i];
            pmIndex[i] = n;
            n += m;
        
    
    for (j = 2; j < 4; j++) 
        pmIndex = mIndex[j];
        n = 0;
        for (i = 0; i < 512; i++)  
            m = pmIndex[i];
            pmIndex[i] = n;
            n += m;
        
    

    pmIndex = mIndex[3];
    for (i = 0; i < count; i++)        /* radix sort */
        u = a[i];
        b[pmIndex[(u >> 26) & 0x1ff]++] = u;
    
    pmIndex = mIndex[2];
    for (i = 0; i < count; i++)  
        u = b[i];
        a[pmIndex[(u >> 35) & 0x1ff]++] = u;
    
    pmIndex = mIndex[1];
    for (i = 0; i < count; i++)  
        u = a[i];
        b[pmIndex[(u >> 44) & 0x3ff]++] = u;
    
    pmIndex = mIndex[0];
    for (i = 0; i < count; i++)  
        u = b[i];
        a[pmIndex[(u >> 54) & 0x3ff]++] = u;
    


更快的方法是先按最高有效位 10 位排序,创建 1024 个 bin,然后对 1024 个 bin 排序,10,9,9 位字段,最低有效位在前。这将加快排序速度,因为 1024 个 bin 中的每一个都适合缓存,从而减少了所有随机访问写入的开销。注意 - aIndex 的大小为 1025,用于跟踪最后一个 bin 的大小。

void RadixSort3(uint64_t *, uint64_t *, size_t);

/* split array into 1024 bins according to most significant 10 bits */
void RadixSort(uint64_t *a, uint64_t *b, size_t count)

uint32_t aIndex[1025] = 0;            /* index array */
uint32_t i, m, n;
    for(i = 0; i < count; i++)          /* generate histogram */
        aIndex[(a[i] >> 54)]++;
    n = 0;                              /* convert to indices */
    for (i = 0; i < 1025; i++)  
        m = aIndex[i];
        aIndex[i] = n;
        n += m;
    
    for(i = 0; i < count; i++)          /* sort by ms 10 bits */
        b[aIndex[a[i]>>54]++] = a[i];
    for(i = 1024; i; i--)               /* restore aIndex */
        aIndex[i] = aIndex[i-1];
    aIndex[0] = 0;
    for(i = 0; i < 1024; i++)           /* radix sort the 1024 bins */
        RadixSort3(&b[aIndex[i]], &a[aIndex[i]], aIndex[i+1]-aIndex[i]);


void RadixSort3(uint64_t *a, uint64_t *b, size_t count)

    uint32_t mIndex[3][1024] =  0 ;   /* index matrix */
    uint32_t * pmIndex;                 /* ptr to row of matrix */
    uint32_t i, j, m, n;
    uint64_t u;

    for (i = 0; i < count; i++)        /* generate histograms */
        u = a[i];
        mIndex[2][(u >> 26) & 0x1ff]++;
        mIndex[1][(u >> 35) & 0x1ff]++;
        mIndex[0][(u >> 44) & 0x3ff]++;
    

    for (j = 0; j < 1; j++)            /* convert to indices */
        pmIndex = mIndex[j];
        n = 0;
        for (i = 0; i < 1024; i++)  
            m = pmIndex[i];
            pmIndex[i] = n;
            n += m;
        
    
    for (j = 1; j < 3; j++) 
        pmIndex = mIndex[j];
        n = 0;
        for (i = 0; i < 512; i++)  
            m = pmIndex[i];
            pmIndex[i] = n;
            n += m;
        
    

    pmIndex = mIndex[2];
    for (i = 0; i < count; i++)        /* radix sort */
        u = a[i];
        b[pmIndex[(u >> 26) & 0x1ff]++] = u;
    
    pmIndex = mIndex[1];
    for (i = 0; i < count; i++)  
        u = b[i];
        a[pmIndex[(u >> 35) & 0x1ff]++] = u;
    
    pmIndex = mIndex[0];
    for (i = 0; i < count; i++)  
        u = a[i];
        b[pmIndex[(u >> 44) & 0x3ff]++] = u;
    

【讨论】:

【参考方案2】:

40 MSB 排序的性能很差,比 48 MSB 版本慢很多。

所以我尝试了一个 [6][64],36 MSB 的版本,它以相对较快的 48 MSB 为蓝本。

我知道我想要 38 位,而不是 36。优化的方法是如何将 XY 位置和该点的属性打包到 uint64_t 中。使用 8k X 和 Y 坐标,XY 占用 13+13 位,数据不超过 64-26=38 位。

当前数据最大为 34 或 35 位,因此 36 应该可以工作。

表演如下:

  Time=  6.104 sec = 30.673%, QSORT_UINT64_ARRAY   , hits=1
  Time=  3.117 sec = 15.663%, RADIX_SORT_UINT64_REG, hits=4, 0.779 sec each
  Time=  2.931 sec = 14.731%, RADIX_SORT_UINT64_40R, hits=4, 0.733 sec each
  Time=  2.269 sec = 11.401%, RADIX_SORT_UINT64_48R, hits=4, 0.567 sec each
  Time=  1.663 sec =  8.359%, RADIX_SORT_UINT64_36R, hits=4, 0.416 sec each < FAST
  Time=  1.516 sec =  7.620%, RADIX_SORT_UINT64_32R, hits=4, 0.379 sec each
  Time=  0.734 sec =  3.689%, RADIX_SORT_64        , hits=1

它比 48 位代码快 27%。

而且,如果 36 位太紧,它看起来应该能够扩展到 [6][7] 以对 42 个 MSB 进行排序!

这是完整的代码:

void radix_sort_r64_36(uint64_t *pData, uint64_t *pTemp, size_t count,
    EV_TIME_STR *tsa)

    size_t mIndex[6][64] =  0 ;  /* index matrix */
    size_t * pmIndex;              /* ptr to row of matrix */
    size_t i, j, m, n;
    uint64_t u;
    if(tsa)  time_event(E_RADIX_SORT_UINT64_36R, tsa, E_TIME_EVENT, 1, 0);  

    // 64 -- 56 48 40 32 24 16  -- 8 bits each
    // 64 -- 58 52 46 40 34 28  -- 6 bits each
    for (i = 0; i < count; i++)        /* generate histograms */
        u = pData[i];                   // Igonores Nibbles 0, 1 & 2
        mIndex[5][(u >> 28) & 0x3F]++;  // N2
        mIndex[4][(u >> 34) & 0x3F]++;  // N3
        mIndex[3][(u >> 40) & 0x3F]++;  // N4
        mIndex[2][(u >> 46) & 0x3F]++;  // N5
        mIndex[1][(u >> 52) & 0x3F]++;  // N6
        mIndex[0][(u >> 58) & 0x3F]++;  // N7
    

    for (j = 0; j < 6; j++)            /* convert to indices */
        pmIndex = mIndex[j];
        n = 0;
        for (i = 0; i < 64; i++) 
            m = pmIndex[i];
            pmIndex[i] = n;
            n += m;
        
    

    for (i = 0; i < count; i++)   /* radix sort */
        u = pData[i];
        pTemp[mIndex[5][(u >> 28) & 0x3F]++] = u;
    
    for (i = 0; i < count; i++)  
        u = pTemp[i];
        pData[mIndex[4][(u >> 34) & 0x3F]++] = u;
    
    for (i = 0; i < count; i++) 
        u = pData[i];
        pTemp[mIndex[3][(u >> 40) & 0x3F]++] = u;
    
    for (i = 0; i < count; i++) 
        u = pTemp[i];
        pData[mIndex[2][(u >> 46) & 0x3F]++] = u;
    
    for (i = 0; i < count; i++) 
        u = pData[i];
        pTemp[mIndex[1][(u >> 52) & 0x3F]++] = u;
    
    for (i = 0; i < count; i++) 
        u = pTemp[i];
        pData[mIndex[0][(u >> 58) & 0x3F]++] = u;
    
  // End Radix_Sort_R64_36().

具有 48 MSB 功能的唯一差异:

Unique lines from "/home/brianp/tmp/radix.sort.36.c":
  01) void radix_sort_r64_36(uint64_t *pData, uint64_t *pTemp, size_t count,
  04) size_t mIndex[6][64] =  0 ;  /* index matrix */
  08) if(tsa)  time_event(E_RADIX_SORT_UINT64_36R, tsa, E_TIME_EVENT, 1, 0);
  11) mIndex[5][(u >> 28) & 0x3F]++;  // N2
  12) mIndex[4][(u >> 34) & 0x3F]++;  // N3
  13) mIndex[3][(u >> 40) & 0x3F]++;  // N4
  14) mIndex[2][(u >> 46) & 0x3F]++;  // N5
  15) mIndex[1][(u >> 52) & 0x3F]++;  // N6
  16) mIndex[0][(u >> 58) & 0x3F]++;  // N7
  22) for (i = 0; i < 64; i++) 
  31) pTemp[mIndex[5][(u >> 28) & 0x3F]++] = u;
  35) pData[mIndex[4][(u >> 34) & 0x3F]++] = u;
  39) pTemp[mIndex[3][(u >> 40) & 0x3F]++] = u;
  43) pData[mIndex[2][(u >> 46) & 0x3F]++] = u;
  47) pTemp[mIndex[1][(u >> 52) & 0x3F]++] = u;
  51) pData[mIndex[0][(u >> 58) & 0x3F]++] = u;

Unique lines from "/home/brianp/tmp/radix.sort.48.c":
  01) void radix_sort_r64_48(uint64_t *pData, uint64_t *pTemp, size_t count,
  04) size_t mIndex[6][256] =  0 ;      /* index matrix */
  08) if(tsa)  time_event(E_RADIX_SORT_UINT64_48R, tsa, E_TIME_EVENT, 1, 0);
  14) mIndex[5][(u >> 16) & 0xff]++;  // B2
  15) mIndex[4][(u >> 24) & 0xff]++;  // B3
  16) mIndex[3][(u >> 32) & 0xff]++;  // B4
  17) mIndex[2][(u >> 40) & 0xff]++;  // B5
  18) mIndex[1][(u >> 48) & 0xff]++;  // B6
  19) mIndex[0][(u >> 56) & 0xff]++;  // B7
  25) for (i = 0; i < 256; i++) 
  34) pTemp[mIndex[5][(u >> 16) & 0xff]++] = u;
  38) pData[mIndex[4][(u >> 24) & 0xff]++] = u;
  42) pTemp[mIndex[3][(u >> 32) & 0xff]++] = u;
  46) pData[mIndex[2][(u >> 40) & 0xff]++] = u;
  50) pTemp[mIndex[1][(u >> 48) & 0xff]++] = u;
  54) pData[mIndex[0][(u >> 56) & 0xff]++] = u;

【讨论】:

【参考方案3】:

为了完整性,6 bin,7 bit/bin 42 MSB Radix Sort 可以正常工作,性能与 48 MSB 和 36 MSB 版本完全一致。

  Time=  6.334 sec = 25.435%, QSORT_UINT64_ARRAY   , hits=1
  Time=  3.519 sec = 14.131%, RADIX_SORT_UINT64_REG, hits=4, 0.880 sec each
  Time=  3.273 sec = 13.145%, RADIX_SORT_UINT64_40R, hits=4, 0.818 sec each < anomaly
  Time=  2.680 sec = 10.764%, RADIX_SORT_UINT64_48R, hits=4, 0.670 sec each
  Time=  2.302 sec =  9.246%, RADIX_SORT_UINT64_42R, hits=4, 0.576 sec each < NEW
  Time=  2.025 sec =  8.132%, RADIX_SORT_UINT64_36R, hits=4, 0.506 sec each
  Time=  1.767 sec =  7.094%, RADIX_SORT_UINT64_32R, hits=4, 0.442 sec each
  Time=  0.955 sec =  3.835%, RADIX_SORT_64        , hits=1

谁能解释为什么 40 MSB 排序比 48 MSB 排序慢得多?

完整代码:

void radix_sort_r64_42(uint64_t *pData, uint64_t *pTemp, size_t count,
    EV_TIME_STR *tsa)

    size_t mIndex[6][128] =  0 ;  /* index matrix */
    size_t * pmIndex;              /* ptr to row of matrix */
    size_t i, j, m, n;
    uint64_t u;
    if(tsa)  time_event(E_RADIX_SORT_UINT64_42R, tsa, E_TIME_EVENT, 1, 0);  

    // 64 -- 56 48 40 32 24 16  -- 8 bits each
    // 64 -- 57 50 43 36 29 22  -- 7 bits each
    // 64 -- 58 52 46 40 34 28  -- 6 bits each
    for (i = 0; i < count; i++)        /* generate histograms */
        u = pData[i];                   // Igonores Nibbles 0, 1 & 2
        mIndex[5][(u >> 22) & 0x7F]++;  // N2
        mIndex[4][(u >> 29) & 0x7F]++;  // N3
        mIndex[3][(u >> 36) & 0x7F]++;  // N4
        mIndex[2][(u >> 43) & 0x7F]++;  // N5
        mIndex[1][(u >> 50) & 0x7F]++;  // N6
        mIndex[0][(u >> 57) & 0x7F]++;  // N7
    

    for (j = 0; j < 6; j++)            /* convert to indices */
        pmIndex = mIndex[j];
        n = 0;
        for (i = 0; i < 128; i++) 
            m = pmIndex[i];
            pmIndex[i] = n;
            n += m;
        
    

    for (i = 0; i < count; i++)   /* radix sort */
        u = pData[i];
        pTemp[mIndex[5][(u >> 22) & 0x7F]++] = u;
    
    for (i = 0; i < count; i++)  
        u = pTemp[i];
        pData[mIndex[4][(u >> 29) & 0x7F]++] = u;
    
    for (i = 0; i < count; i++) 
        u = pData[i];
        pTemp[mIndex[3][(u >> 36) & 0x7F]++] = u;
    
    for (i = 0; i < count; i++) 
        u = pTemp[i];
        pData[mIndex[2][(u >> 43) & 0x7F]++] = u;
    
    for (i = 0; i < count; i++) 
        u = pData[i];
        pTemp[mIndex[1][(u >> 50) & 0x7F]++] = u;
    
    for (i = 0; i < count; i++) 
        u = pTemp[i];
        pData[mIndex[0][(u >> 57) & 0x7F]++] = u;
    
  // End Radix_Sort_R64_42().

添加 36 MSB 与 42 MSB 的差异版本

Unique lines from "~/tmp/radix.sort.36.c":
  01) void radix_sort_r64_36(uint64_t *pData, uint64_t *pTemp, size_t count,
  04) size_t mIndex[6][64] =  0 ;  /* index matrix */
  11) mIndex[5][(u >> 28) & 0x3F]++;  // N2
  12) mIndex[4][(u >> 34) & 0x3F]++;  // N3
  13) mIndex[3][(u >> 40) & 0x3F]++;  // N4
  14) mIndex[2][(u >> 46) & 0x3F]++;  // N5
  15) mIndex[1][(u >> 52) & 0x3F]++;  // N6
  16) mIndex[0][(u >> 58) & 0x3F]++;  // N7
  22) for (i = 0; i < 64; i++) 
  31) pTemp[mIndex[5][(u >> 28) & 0x3F]++] = u;
  35) pData[mIndex[4][(u >> 34) & 0x3F]++] = u;
  39) pTemp[mIndex[3][(u >> 40) & 0x3F]++] = u;
  43) pData[mIndex[2][(u >> 46) & 0x3F]++] = u;
  47) pTemp[mIndex[1][(u >> 52) & 0x3F]++] = u;
  51) pData[mIndex[0][(u >> 58) & 0x3F]++] = u;

19 Unique lines from "~/tmp/radix.sort.42.c":
  01) void radix_sort_r64_42(uint64_t *pData, uint64_t *pTemp, size_t count,
  04) size_t mIndex[6][128] =  0 ;  /* index matrix */
  10) // 64 -- 56 48 40 32 24 16  -- 8 bits each
  11) // 64 -- 57 50 43 36 29 22  -- 7 bits each
  12) // 64 -- 58 52 46 40 34 28  -- 6 bits each
  15) mIndex[5][(u >> 22) & 0x7F]++;  // N2
  16) mIndex[4][(u >> 29) & 0x7F]++;  // N3
  17) mIndex[3][(u >> 36) & 0x7F]++;  // N4
  18) mIndex[2][(u >> 43) & 0x7F]++;  // N5
  19) mIndex[1][(u >> 50) & 0x7F]++;  // N6
  20) mIndex[0][(u >> 57) & 0x7F]++;  // N7
  26) for (i = 0; i < 128; i++) 
  35) pTemp[mIndex[5][(u >> 22) & 0x7F]++] = u;
  39) pData[mIndex[4][(u >> 29) & 0x7F]++] = u;
  43) pTemp[mIndex[3][(u >> 36) & 0x7F]++] = u;
  47) pData[mIndex[2][(u >> 43) & 0x7F]++] = u;
  51) pTemp[mIndex[1][(u >> 50) & 0x7F]++] = u;
  55) pData[mIndex[0][(u >> 57) & 0x3F]++] = u;

【讨论】:

【参考方案4】:

另外2个排序算法测试结果:

  Time=  6.208 sec = 21.838%, QSORT_UINT64_ARRAY   , hits=1
  Time=  3.358 sec = 11.813%, RADIX_SORT_UINT64_REG, hits=4, 0.840 sec each
  Time=  2.525 sec =  8.884%, RADIX_SORT_UI64_AA99 , hits=4, 0.631 sec each <NEW
  Time=  2.509 sec =  8.825%, RADIX_SORT_UINT64_48R, hits=4, 0.627 sec each
  Time=  2.461 sec =  8.658%, RADIX_SORT_UI64_1024 , hits=4, 0.615 sec each <NEW
  Time=  2.223 sec =  7.822%, RADIX_SORT_UINT64_42R, hits=4, 0.556 sec each
  Time=  2.215 sec =  7.791%, RADIX_SORT_UI64_40_85, hits=4, 0.554 sec each
  Time=  1.930 sec =  6.788%, RADIX_SORT_UINT64_36R, hits=4, 0.482 sec each
  Time=  1.710 sec =  6.014%, RADIX_SORT_UINT64_32R, hits=4, 0.427 sec each
  Time=  0.915 sec =  3.220%, COMP_UINT64_ARRAYS   , hits=32, 0.029 sec each

另一个没有 Firefox 占用系统的运行:

  Time=  6.156 sec = 23.199%, QSORT_UINT64_ARRAY   , hits=1
  Time=  2.993 sec = 11.277%, RADIX_SORT_UINT64_REG, hits=4, 0.748 sec each
  Time=  2.409 sec =  9.077%, RADIX_SORT_UI64_AA99 , hits=4, 0.602 sec each < NEW
  Time=  2.330 sec =  8.778%, RADIX_SORT_UI64_1024 , hits=4, 0.582 sec each < NEW
  Time=  2.241 sec =  8.443%, RADIX_SORT_UINT64_48R, hits=4, 0.560 sec each
  Time=  2.124 sec =  8.002%, RADIX_SORT_UI64_40_85, hits=4, 0.531 sec each
  Time=  1.982 sec =  7.468%, RADIX_SORT_UINT64_42R, hits=4, 0.495 sec each
  Time=  1.725 sec =  6.499%, RADIX_SORT_UINT64_36R, hits=4, 0.431 sec each
  Time=  1.507 sec =  5.677%, RADIX_SORT_UINT64_32R, hits=4, 0.377 sec each
  Time=  0.889 sec =  3.348%, COMP_UINT64_ARRAYS   , hits=32, 0.028 sec each

RADIX_SORT_UI64_AA99 在 4 个通道中使用 bin 位 [10, 10, 9, 9]。

RADIX_SORT_UI64_1024 首先按最高有效 10 位排序。

RADIX_SORT_UINT64_42R 使用 6 个 bin,每个 bin 有 7 位

注意决斗 40 MSB、8 bin 5 bit RADIX_SORT_UI64_40_85 和 42 MSB、7 bin 6 bit RADIX_SORT_UINT64_42R 的相对性能。 42 MSB 排序通常要快得多,尽管 40 位有时会将其排除在外。

编译器是 GCC,针对速度进行了优化:

gcc -Ofast -ffast-math -m64 -march=native -funroll-loops -fopenmp -flto -finline-functions -Wuninitialized  ~/bin/pb.c  -lm  -o  ~/bin/pb_a 

有更好的编译器选项吗?

gcc 版本 7.4.1 20190905 [gcc-7-branch 修订版 275407] (SUSE Linux)

===========================

完整代码:

// =============================================================================
// Sort with bin bits 10, 10, 9,9
// From code by rcgldr *** Feb 8, 2020
void RadixSort_aa99(uint64_t *a, uint64_t *b, size_t count, EV_TIME_STR *tsa)

uint32_t mIndex[4][1024] =  0 ;  /* index matrix */
uint32_t * pmIndex;                /* ptr to row of matrix */
uint32_t i, j, m, n;
uint64_t u;

if(tsa)  time_event(E_RADIX_SORT_UI64_AA99, tsa, E_TIME_EVENT, 1, 0);  
for (i = 0; i < count; i++)        /* generate histograms */
    u = a[i];
    mIndex[3][(u >> 26) & 0x1ff]++;
    mIndex[2][(u >> 35) & 0x1ff]++;
    mIndex[1][(u >> 44) & 0x3ff]++;
    mIndex[0][(u >> 54) & 0x3ff]++;


for (j = 0; j < 2; j++)            /* convert to indices */
    pmIndex = mIndex[j];
    n = 0;
    for (i = 0; i < 1024; i++)  
        m = pmIndex[i];
        pmIndex[i] = n;
        n += m;
    

for (j = 2; j < 4; j++) 
    pmIndex = mIndex[j];
    n = 0;
    for (i = 0; i < 512; i++)  
        m = pmIndex[i];
        pmIndex[i] = n;
        n += m;
    


pmIndex = mIndex[3];
for (i = 0; i < count; i++)        /* radix sort */
    u = a[i];
    b[pmIndex[(u >> 26) & 0x1ff]++] = u;

pmIndex = mIndex[2];
for (i = 0; i < count; i++)  
    u = b[i];
    a[pmIndex[(u >> 35) & 0x1ff]++] = u;

pmIndex = mIndex[1];
for (i = 0; i < count; i++)  
    u = a[i];
    b[pmIndex[(u >> 44) & 0x3ff]++] = u;

pmIndex = mIndex[0];
for (i = 0; i < count; i++)  
    u = b[i];
    a[pmIndex[(u >> 54) & 0x3ff]++] = u;




// =============================================================================
/* split array into 1024 bins according to most significant 10 bits */
void RadixSort_1024(uint64_t *a, uint64_t *b, size_t count, EV_TIME_STR *tsa)

uint32_t aIndex[1025] = 0;            /* index array */
uint32_t i, m, n;
if(tsa)  time_event(E_RADIX_SORT_UI64_1024, tsa, E_TIME_EVENT, 1, 0);  
for(i = 0; i < count; i++)          /* generate histogram */
    aIndex[(a[i] >> 54)]++;
n = 0;                              /* convert to indices */
for (i = 0; i < 1025; i++)  
    m = aIndex[i];
    aIndex[i] = n;
    n += m;

for(i = 0; i < count; i++)          /* sort by ms 10 bits */
    b[aIndex[a[i]>>54]++] = a[i];
for(i = 1024; i; i--)               /* restore aIndex */
    aIndex[i] = aIndex[i-1];
aIndex[0] = 0;
for(i = 0; i < 1024; i++)           /* radix sort the 1024 bins */
    RadixSort3(&b[aIndex[i]], &a[aIndex[i]], aIndex[i+1]-aIndex[i]);


void RadixSort3(uint64_t *a, uint64_t *b, size_t count)

uint32_t mIndex[3][1024] =  0 ;   /* index matrix */
uint32_t * pmIndex;                 /* ptr to row of matrix */
uint32_t i, j, m, n;
uint64_t u;

for (i = 0; i < count; i++)        /* generate histograms */
    u = a[i];
    mIndex[2][(u >> 26) & 0x1ff]++;
    mIndex[1][(u >> 35) & 0x1ff]++;
    mIndex[0][(u >> 44) & 0x3ff]++;


for (j = 0; j < 1; j++)            /* convert to indices */
    pmIndex = mIndex[j];
    n = 0;
    for (i = 0; i < 1024; i++)  
        m = pmIndex[i];
        pmIndex[i] = n;
        n += m;
    

for (j = 1; j < 3; j++) 
    pmIndex = mIndex[j];
    n = 0;
    for (i = 0; i < 512; i++)  
        m = pmIndex[i];
        pmIndex[i] = n;
        n += m;
    


pmIndex = mIndex[2];
for (i = 0; i < count; i++)        /* radix sort */
    u = a[i];
    b[pmIndex[(u >> 26) & 0x1ff]++] = u;

pmIndex = mIndex[1];
for (i = 0; i < count; i++)  
    u = b[i];
    a[pmIndex[(u >> 35) & 0x1ff]++] = u;

pmIndex = mIndex[0];
for (i = 0; i < count; i++)  
    u = a[i];
    b[pmIndex[(u >> 44) & 0x3ff]++] = u;


【讨论】:

设置大小为 36.2M uint64_t 和 100% 不同,使用 mrand48() 生成两次,

以上是关于在 uint64 中对 38 位进行基数排序?的主要内容,如果未能解决你的问题,请参考以下文章

基数排序 - 主位优先

对 32 位数字使用基数排序

使用按位运算进行基数排序

#19 基数排序(Radix Sort)

基数排序及其并行化

9经典算法基数排序