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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
/*
* Copyright © 2012 Intel Corporation
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library. If not, see <http://www.gnu.org/licenses/>.
*
*/
/**
* \file sampler.cpp
*
*/
#include "sampler.hpp"
#include "context.hpp"
#include "ocl_common_defines.h"
namespace gbe {
namespace ir {
#ifdef GBE_COMPILER_AVAILABLE
uint8_t SamplerSet::appendReg(uint32_t key, Context *ctx) {
uint8_t samplerSlot = samplerMap.size();
samplerMap.insert(std::make_pair(key, samplerSlot));
return samplerSlot;
}
uint8_t SamplerSet::append(uint32_t samplerValue, Context *ctx)
{
auto it = samplerMap.find(samplerValue);
if (it != samplerMap.end())
return it->second;
// This register is just used as a key.
return appendReg(samplerValue, ctx);
}
#define SAMPLER_ID(id) ((id << __CLK_SAMPLER_ARG_BASE) | __CLK_SAMPLER_ARG_KEY_BIT)
uint8_t SamplerSet::append(Register samplerReg, Context *ctx)
{
ir::FunctionArgument *arg = ctx->getFunction().getArg(samplerReg);
GBE_ASSERT(arg != NULL);
// XXX As LLVM 3.2/3.1 doesn't have a new data type for the sampler_t, we have to fix up the argument
// type here. Once we switch to the LLVM and use the new data type sampler_t, we can remove this
// work around.
arg->type = ir::FunctionArgument::SAMPLER;
arg->info.typeName = "sampler_t";
int32_t id = ctx->getFunction().getArgID(arg);
GBE_ASSERT(id < (1 << __CLK_SAMPLER_ARG_BITS));
auto it = samplerMap.find(SAMPLER_ID(id));
if (it != samplerMap.end()) {
return it->second;
}
return appendReg(SAMPLER_ID(id), ctx);
}
#endif
#define OUT_UPDATE_SZ(elt) SERIALIZE_OUT(elt, outs, ret_size)
#define IN_UPDATE_SZ(elt) DESERIALIZE_IN(elt, ins, total_size)
/*! Implements the serialization. */
size_t SamplerSet::serializeToBin(std::ostream& outs) {
size_t ret_size = 0;
OUT_UPDATE_SZ(magic_begin);
OUT_UPDATE_SZ(samplerMap.size());
for (auto iter : samplerMap) {
OUT_UPDATE_SZ(iter.first);
OUT_UPDATE_SZ(iter.second);
}
OUT_UPDATE_SZ(magic_end);
OUT_UPDATE_SZ(ret_size);
return ret_size;
}
size_t SamplerSet::deserializeFromBin(std::istream& ins) {
size_t total_size = 0;
uint32_t magic;
size_t sampler_map_sz = 0;
IN_UPDATE_SZ(magic);
if (magic != magic_begin)
return 0;
IN_UPDATE_SZ(sampler_map_sz);
for (size_t i = 0; i < sampler_map_sz; i++) {
uint32_t key;
uint32_t slot;
IN_UPDATE_SZ(key);
IN_UPDATE_SZ(slot);
samplerMap.insert(std::make_pair(key, slot));
}
IN_UPDATE_SZ(magic);
if (magic != magic_end)
return 0;
size_t total_bytes;
IN_UPDATE_SZ(total_bytes);
if (total_bytes + sizeof(total_size) != total_size)
return 0;
return total_size;
}
void SamplerSet::printStatus(int indent, std::ostream& outs) {
using namespace std;
string spaces = indent_to_str(indent);
string spaces_nl = indent_to_str(indent + 4);
outs << spaces << "------------ Begin SamplerSet ------------" << "\n";
outs << spaces_nl << " SamplerSet Map: [index, sampler_reg, sampler_slot]\n";
outs << spaces_nl << " samplerMap size: " << samplerMap.size() << "\n";
for (auto iter : samplerMap) {
outs << spaces_nl << " [" << iter.first << ", "
<< iter.second << "]\n";
}
outs << spaces << "------------- End SamplerSet -------------" << "\n";
}
} /* namespace ir */
} /* namespace gbe */
|
