1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
// Copyright (c) 2024, Intel Corporation
// SPDX-License-Identifier: BSD-3-Clause
// For easier checking on the C++ side:
// The value of PERMUTE_VAL should match the value of 'perm' parameter defined in the C++ code.
// This ensures that during runtime, the shuffle index simplifies to just programIndex.
#define PERMUTE_VAL 1
template <typename T> inline void Shuffle1(uniform T a[], uniform T ret[], int perm, uniform int count) {
foreach (i = 0 ... count) {
ret[i] = shuffle(a[i], programIndex + perm - PERMUTE_VAL);
}
}
template <typename T> inline void Shuffle2(uniform T a[], uniform T b[], uniform T ret[], int perm, uniform int count) {
foreach (i = 0 ... count) {
ret[i] = shuffle(a[i], b[i], programIndex + perm - PERMUTE_VAL);
}
}
#define DEFINE_SHUFFLE1(TYPE, TYPE_NAME) \
export void Shuffle1_##TYPE_NAME(uniform TYPE a[], uniform TYPE ret[], uniform int perm, uniform int count) { \
Shuffle1<TYPE>(a, ret, perm, count); \
}
#define DEFINE_SHUFFLE2(TYPE, TYPE_NAME) \
export void Shuffle2_##TYPE_NAME(uniform TYPE a[], uniform TYPE b[], uniform TYPE ret[], uniform int perm, \
uniform int count) { \
Shuffle2<TYPE>(a, b, ret, perm, count); \
}
// Define Shuffle1 functions for each type
DEFINE_SHUFFLE1(int8, int8)
DEFINE_SHUFFLE1(int16, int16)
DEFINE_SHUFFLE1(int, int)
DEFINE_SHUFFLE1(float, float)
DEFINE_SHUFFLE1(int64, int64)
DEFINE_SHUFFLE1(double, double)
// Define Shuffle2 functions for each type
DEFINE_SHUFFLE2(int8, int8)
DEFINE_SHUFFLE2(int16, int16)
DEFINE_SHUFFLE2(int, int)
DEFINE_SHUFFLE2(float, float)
DEFINE_SHUFFLE2(int64, int64)
DEFINE_SHUFFLE2(double, double)
|
