2018-03-13 09:46:55 +00:00
|
|
|
#pragma clang diagnostic ignored "-Wmissing-prototypes"
|
2019-08-14 15:09:39 +00:00
|
|
|
#pragma clang diagnostic ignored "-Wmissing-braces"
|
|
|
|
#pragma clang diagnostic ignored "-Wunused-variable"
|
2018-03-13 09:46:55 +00:00
|
|
|
|
|
|
|
#include <metal_stdlib>
|
|
|
|
#include <simd/simd.h>
|
2019-08-14 15:09:39 +00:00
|
|
|
|
|
|
|
template <typename T, size_t Num>
|
|
|
|
struct unsafe_array
|
|
|
|
{
|
|
|
|
T __Elements[Num ? Num : 1];
|
|
|
|
|
|
|
|
constexpr size_t size() const thread { return Num; }
|
|
|
|
constexpr size_t max_size() const thread { return Num; }
|
|
|
|
constexpr bool empty() const thread { return Num == 0; }
|
|
|
|
|
|
|
|
constexpr size_t size() const device { return Num; }
|
|
|
|
constexpr size_t max_size() const device { return Num; }
|
|
|
|
constexpr bool empty() const device { return Num == 0; }
|
|
|
|
|
|
|
|
constexpr size_t size() const constant { return Num; }
|
|
|
|
constexpr size_t max_size() const constant { return Num; }
|
|
|
|
constexpr bool empty() const constant { return Num == 0; }
|
|
|
|
|
|
|
|
constexpr size_t size() const threadgroup { return Num; }
|
|
|
|
constexpr size_t max_size() const threadgroup { return Num; }
|
|
|
|
constexpr bool empty() const threadgroup { return Num == 0; }
|
|
|
|
|
|
|
|
thread T &operator[](size_t pos) thread
|
|
|
|
{
|
|
|
|
return __Elements[pos];
|
|
|
|
}
|
|
|
|
constexpr const thread T &operator[](size_t pos) const thread
|
|
|
|
{
|
|
|
|
return __Elements[pos];
|
|
|
|
}
|
|
|
|
|
|
|
|
device T &operator[](size_t pos) device
|
|
|
|
{
|
|
|
|
return __Elements[pos];
|
|
|
|
}
|
|
|
|
constexpr const device T &operator[](size_t pos) const device
|
|
|
|
{
|
|
|
|
return __Elements[pos];
|
|
|
|
}
|
|
|
|
|
|
|
|
constexpr const constant T &operator[](size_t pos) const constant
|
|
|
|
{
|
|
|
|
return __Elements[pos];
|
|
|
|
}
|
|
|
|
|
|
|
|
threadgroup T &operator[](size_t pos) threadgroup
|
|
|
|
{
|
|
|
|
return __Elements[pos];
|
|
|
|
}
|
|
|
|
constexpr const threadgroup T &operator[](size_t pos) const threadgroup
|
|
|
|
{
|
|
|
|
return __Elements[pos];
|
|
|
|
}
|
|
|
|
};
|
2018-03-13 09:46:55 +00:00
|
|
|
|
|
|
|
using namespace metal;
|
|
|
|
|
|
|
|
// Implementation of the GLSL findLSB() function
|
|
|
|
template<typename T>
|
2019-08-14 15:09:39 +00:00
|
|
|
<<<<<<< HEAD
|
2019-08-29 21:39:06 +00:00
|
|
|
inline T spvFindLSB(T x)
|
2019-08-14 15:09:39 +00:00
|
|
|
=======
|
|
|
|
static inline __attribute__((always_inline))
|
|
|
|
T findLSB(T x)
|
|
|
|
>>>>>>> 22755a0c... Update the Metal shaders to account for changes in the shader compilation.
|
2018-03-13 09:46:55 +00:00
|
|
|
{
|
|
|
|
return select(ctz(x), T(-1), x == T(0));
|
|
|
|
}
|
|
|
|
|
|
|
|
// Implementation of the signed GLSL findMSB() function
|
|
|
|
template<typename T>
|
2019-08-14 15:09:39 +00:00
|
|
|
<<<<<<< HEAD
|
2019-08-29 21:39:06 +00:00
|
|
|
inline T spvFindSMSB(T x)
|
2019-08-14 15:09:39 +00:00
|
|
|
=======
|
|
|
|
static inline __attribute__((always_inline))
|
|
|
|
T findSMSB(T x)
|
|
|
|
>>>>>>> 22755a0c... Update the Metal shaders to account for changes in the shader compilation.
|
2018-03-13 09:46:55 +00:00
|
|
|
{
|
|
|
|
T v = select(x, T(-1) - x, x < T(0));
|
|
|
|
return select(clz(T(0)) - (clz(v) + T(1)), T(-1), v == T(0));
|
|
|
|
}
|
|
|
|
|
|
|
|
// Implementation of the unsigned GLSL findMSB() function
|
|
|
|
template<typename T>
|
2019-08-14 15:09:39 +00:00
|
|
|
<<<<<<< HEAD
|
2019-08-29 21:39:06 +00:00
|
|
|
inline T spvFindUMSB(T x)
|
2019-08-14 15:09:39 +00:00
|
|
|
=======
|
|
|
|
static inline __attribute__((always_inline))
|
|
|
|
T findUMSB(T x)
|
|
|
|
>>>>>>> 22755a0c... Update the Metal shaders to account for changes in the shader compilation.
|
2018-03-13 09:46:55 +00:00
|
|
|
{
|
|
|
|
return select(clz(T(0)) - (clz(x) + T(1)), T(-1), x == T(0));
|
|
|
|
}
|
|
|
|
|
|
|
|
kernel void main0()
|
|
|
|
{
|
|
|
|
int signed_value = 0;
|
|
|
|
uint unsigned_value = 0u;
|
2019-08-26 09:01:14 +00:00
|
|
|
int s = extract_bits(signed_value, uint(5), uint(20));
|
|
|
|
uint u = extract_bits(unsigned_value, uint(6), uint(21));
|
|
|
|
s = insert_bits(s, 40, uint(5), uint(4));
|
|
|
|
u = insert_bits(u, 60u, uint(5), uint(4));
|
2018-03-13 09:46:55 +00:00
|
|
|
u = reverse_bits(u);
|
|
|
|
s = reverse_bits(s);
|
2019-08-26 09:01:14 +00:00
|
|
|
int v0 = int(popcount(u));
|
2018-03-13 09:46:55 +00:00
|
|
|
int v1 = popcount(s);
|
2019-07-12 08:57:56 +00:00
|
|
|
int v2 = int(spvFindUMSB(u));
|
|
|
|
int v3 = spvFindSMSB(s);
|
|
|
|
int v4 = int(spvFindLSB(u));
|
|
|
|
int v5 = spvFindLSB(s);
|
2018-03-13 09:46:55 +00:00
|
|
|
}
|
|
|
|
|