gpt4 book ai didi

CUDA 推力 : how to realize "partition" that supports "stencil"?

转载 作者:行者123 更新时间:2023-12-01 04:35:04 29 4
gpt4 key购买 nike

假设有一个整数数组:

A[]={2, 2, 9, 8, 5, 7, 0, 6}

和模板:

B[]={1, 0, 0, 1, 1, 1, 0, 1}

我的问题是我们如何根据 B[] 重新排列 A[] 使得如果 B[i]==1, B[j]==0, 那么 A[i] 将保证先于 A[j ] 在新数组中,它应该如下所示:

C[]={2, 8, 5, 7, 6, 2, 9, 0}

PS:我找到了 "partition"函数几乎就是答案,只是它只支持谓词。有什么解决方法吗?

非常感谢任何提示!

最佳答案

这可以使用 thrust::stable_sort_by_key() 来实现.

关于CUDA 推力 : how to realize "partition" that supports "stencil"?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/12047961/

29 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com