Мне нужно вычислить медиану массива размера p внутри ядра CUDA (в моем случае p мало, например, p = 10). Я использую алгоритм O(p^2) из-за его простоты, но ценой снижения производительности.
Есть ли «функция» для эффективного нахождения медианы, которую я могу вызвать внутри ядра CUDA?
Я знаю, что могу реализовать алгоритм выбора, но мне нужна функция и/или проверенный код.
Спасибо!
Вот несколько советов:
Есть много других оптимизаций, которые вы можете сделать. Обязательно прочтите документы CUDA, особенно Руководство по программированию и Руководство по передовому опыту. Если вы действительно хотите добиться высокой производительности, не забудьте внимательно изучить профайлер CUDA, например Visual Profiler.
Даже в одном потоке можно отсортировать массив и выбрать значение посередине в O(p*log(p)), из-за чего O(p^2) выглядит чрезмерным. Если в вашем распоряжении есть потоки p, также можно отсортировать массив со скоростью O(log(p)), хотя это может быть не самым быстрым решением для малых p. См. верхний ответ здесь:
Какой алгоритм параллельной сортировки имеет наилучшую среднюю производительность?< /а>
p, рассматривали ли вы возможность использования шаблонной функции, использующей минимальную сеть сортировки? - person njuffa schedule 30.08.2013p, вероятно, указывает на то, что вы должны написать свой собственный код, как уже предлагали другие. Если вы хотите ознакомиться с некоторыми базовыми примерами кодов, не забудьте просмотреть различные коды сортировки в образцы cuda, а также CUB. - person Robert Crovella schedule 30.08.2013