0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-11-18 16:00:57 +01:00
construct/include/ircd/simt/sort.h

39 lines
1 KiB
C
Raw Normal View History

2021-04-02 21:51:59 +02:00
// Matrix Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2021 Jason Volk <jason@zemos.net>
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice is present in all copies. The
// full license for this software is available in the LICENSE file.
#pragma once
#define HAVE_IRCD_SIMT_SORT_H
#ifdef __OPENCL_C_VERSION__
2021-04-02 21:51:59 +02:00
/// Sort indices in `idx` which point to values contained in `val`.
inline void
ircd_simt_sort_idx16_flldr(__local ushort *const idx,
__global const float *const val)
2021-04-02 21:51:59 +02:00
{
const uint
li = get_local_id(0),
ln = get_local_size(0);
2021-04-02 21:51:59 +02:00
for(uint stride = ln >> 1; stride > 0; stride >>= 1)
{
barrier(CLK_LOCAL_MEM_FENCE);
if(li >= stride || val[idx[li]] >= val[idx[li + stride]])
continue;
2021-04-02 21:51:59 +02:00
const ushort
ours = idx[li],
theirs = idx[li + stride];
2021-04-02 21:51:59 +02:00
idx[li] = theirs;
idx[li + stride] = ours;
2021-04-02 21:51:59 +02:00
}
}
#endif