mirror of
https://github.com/matrix-construct/construct
synced 2024-11-29 10:12:39 +01:00
40 lines
1.1 KiB
C
40 lines
1.1 KiB
C
// 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_NORM_H
|
|
|
|
#ifdef __OPENCL_C_VERSION__
|
|
/// Normalize the input, placing the result in possibly overlapping output.
|
|
/// This procedure requires an additional temporary buffer.
|
|
inline void
|
|
ircd_simt_math_norm_f4lldr(__local float4 *const out,
|
|
__local const float4 *const in,
|
|
__local float4 *const restrict tmp)
|
|
{
|
|
const uint
|
|
li = get_local_id(0),
|
|
ln = get_local_size(0);
|
|
|
|
ircd_simt_math_mean_f4lldr(tmp, in);
|
|
|
|
const float4
|
|
sub_mean = in[li] - tmp[li];
|
|
|
|
tmp[li] = pow(sub_mean, 2);
|
|
ircd_simt_math_mean_f4lldr(out, tmp);
|
|
|
|
const float4
|
|
epsilon = 0.00001f,
|
|
s = sqrt(out[li] + epsilon);
|
|
|
|
out[li] = sub_mean / s;
|
|
}
|
|
#endif
|