clover: Pass corresponding module::argument to kernel::argument::bind().
And remove size information from most kernel::argument derived
classes, it's no longer going to be necessary.
Cc: mesa-stable@lists.freedesktop.org
Reviewed-by: Tom Stellard <thomas.stellard@amd.com>
diff --git a/src/gallium/state_trackers/clover/core/kernel.cpp b/src/gallium/state_trackers/clover/core/kernel.cpp
index 68e4137..9e7c088 100644
--- a/src/gallium/state_trackers/clover/core/kernel.cpp
+++ b/src/gallium/state_trackers/clover/core/kernel.cpp
@@ -28,25 +28,25 @@
_cl_kernel::_cl_kernel(clover::program &prog,
const std::string &name,
- const std::vector<clover::module::argument> &args) :
+ const std::vector<clover::module::argument> &margs) :
prog(prog), __name(name), exec(*this) {
- for (auto arg : args) {
- if (arg.type == module::argument::scalar)
- this->args.emplace_back(new scalar_argument(arg.size));
- else if (arg.type == module::argument::global)
- this->args.emplace_back(new global_argument(arg.size));
- else if (arg.type == module::argument::local)
- this->args.emplace_back(new local_argument());
- else if (arg.type == module::argument::constant)
- this->args.emplace_back(new constant_argument());
- else if (arg.type == module::argument::image2d_rd ||
- arg.type == module::argument::image3d_rd)
- this->args.emplace_back(new image_rd_argument());
- else if (arg.type == module::argument::image2d_wr ||
- arg.type == module::argument::image3d_wr)
- this->args.emplace_back(new image_wr_argument());
- else if (arg.type == module::argument::sampler)
- this->args.emplace_back(new sampler_argument());
+ for (auto marg : margs) {
+ if (marg.type == module::argument::scalar)
+ args.emplace_back(new scalar_argument(marg.size));
+ else if (marg.type == module::argument::global)
+ args.emplace_back(new global_argument);
+ else if (marg.type == module::argument::local)
+ args.emplace_back(new local_argument);
+ else if (marg.type == module::argument::constant)
+ args.emplace_back(new constant_argument);
+ else if (marg.type == module::argument::image2d_rd ||
+ marg.type == module::argument::image3d_rd)
+ args.emplace_back(new image_rd_argument);
+ else if (marg.type == module::argument::image2d_wr ||
+ marg.type == module::argument::image3d_wr)
+ args.emplace_back(new image_wr_argument);
+ else if (marg.type == module::argument::sampler)
+ args.emplace_back(new sampler_argument);
else
throw error(CL_INVALID_KERNEL_DEFINITION);
}
@@ -129,7 +129,6 @@
return prog.binaries().find(&q.dev)->second;
}
-
_cl_kernel::exec_context::exec_context(clover::kernel &kern) :
kern(kern), q(NULL), mem_local(0), st(NULL) {
}
@@ -143,8 +142,12 @@
_cl_kernel::exec_context::bind(clover::command_queue *__q) {
std::swap(q, __q);
- for (auto &arg : kern.args)
- arg->bind(*this);
+ // Bind kernel arguments.
+ auto margs = kern.module(*q).sym(kern.name()).args;
+ for_each([=](std::unique_ptr<kernel::argument> &karg,
+ const module::argument &marg) {
+ karg->bind(*this, marg);
+ }, kern.args.begin(), kern.args.end(), margs.begin());
// Create a new compute state if anything changed.
if (!st || q != __q ||
@@ -176,8 +179,7 @@
mem_local = 0;
}
-_cl_kernel::argument::argument(size_t size) :
- __size(size), __set(false) {
+_cl_kernel::argument::argument() : __set(false) {
}
bool
@@ -190,13 +192,12 @@
return 0;
}
-_cl_kernel::scalar_argument::scalar_argument(size_t size) :
- argument(size) {
+_cl_kernel::scalar_argument::scalar_argument(size_t size) : size(size) {
}
void
_cl_kernel::scalar_argument::set(size_t size, const void *value) {
- if (size != __size)
+ if (size != this->size)
throw error(CL_INVALID_ARG_SIZE);
v = { (uint8_t *)value, (uint8_t *)value + size };
@@ -204,7 +205,8 @@
}
void
-_cl_kernel::scalar_argument::bind(exec_context &ctx) {
+_cl_kernel::scalar_argument::bind(exec_context &ctx,
+ const clover::module::argument &marg) {
ctx.input.insert(ctx.input.end(), v.begin(), v.end());
}
@@ -212,10 +214,6 @@
_cl_kernel::scalar_argument::unbind(exec_context &ctx) {
}
-_cl_kernel::global_argument::global_argument(size_t size) :
- argument(size) {
-}
-
void
_cl_kernel::global_argument::set(size_t size, const void *value) {
if (size != sizeof(cl_mem))
@@ -229,11 +227,12 @@
}
void
-_cl_kernel::global_argument::bind(exec_context &ctx) {
+_cl_kernel::global_argument::bind(exec_context &ctx,
+ const clover::module::argument &marg) {
size_t offset = ctx.input.size();
size_t idx = ctx.g_buffers.size();
- ctx.input.resize(offset + __size);
+ ctx.input.resize(offset + marg.size);
ctx.g_buffers.resize(idx + 1);
ctx.g_buffers[idx] = obj->resource(ctx.q).pipe;
@@ -246,10 +245,6 @@
_cl_kernel::global_argument::unbind(exec_context &ctx) {
}
-_cl_kernel::local_argument::local_argument() :
- argument(sizeof(uint32_t)) {
-}
-
size_t
_cl_kernel::local_argument::storage() const {
return __storage;
@@ -265,7 +260,8 @@
}
void
-_cl_kernel::local_argument::bind(exec_context &ctx) {
+_cl_kernel::local_argument::bind(exec_context &ctx,
+ const clover::module::argument &marg) {
size_t offset = ctx.input.size();
size_t ptr = ctx.mem_local;
@@ -279,10 +275,6 @@
_cl_kernel::local_argument::unbind(exec_context &ctx) {
}
-_cl_kernel::constant_argument::constant_argument() :
- argument(sizeof(uint32_t)) {
-}
-
void
_cl_kernel::constant_argument::set(size_t size, const void *value) {
if (size != sizeof(cl_mem))
@@ -296,7 +288,8 @@
}
void
-_cl_kernel::constant_argument::bind(exec_context &ctx) {
+_cl_kernel::constant_argument::bind(exec_context &ctx,
+ const clover::module::argument &marg) {
size_t offset = ctx.input.size();
size_t idx = ctx.resources.size();
@@ -312,10 +305,6 @@
obj->resource(ctx.q).unbind_surface(*ctx.q, st);
}
-_cl_kernel::image_rd_argument::image_rd_argument() :
- argument(sizeof(uint32_t)) {
-}
-
void
_cl_kernel::image_rd_argument::set(size_t size, const void *value) {
if (size != sizeof(cl_mem))
@@ -329,7 +318,8 @@
}
void
-_cl_kernel::image_rd_argument::bind(exec_context &ctx) {
+_cl_kernel::image_rd_argument::bind(exec_context &ctx,
+ const clover::module::argument &marg) {
size_t offset = ctx.input.size();
size_t idx = ctx.sviews.size();
@@ -345,10 +335,6 @@
obj->resource(ctx.q).unbind_sampler_view(*ctx.q, st);
}
-_cl_kernel::image_wr_argument::image_wr_argument() :
- argument(sizeof(uint32_t)) {
-}
-
void
_cl_kernel::image_wr_argument::set(size_t size, const void *value) {
if (size != sizeof(cl_mem))
@@ -362,7 +348,8 @@
}
void
-_cl_kernel::image_wr_argument::bind(exec_context &ctx) {
+_cl_kernel::image_wr_argument::bind(exec_context &ctx,
+ const clover::module::argument &marg) {
size_t offset = ctx.input.size();
size_t idx = ctx.resources.size();
@@ -378,10 +365,6 @@
obj->resource(ctx.q).unbind_surface(*ctx.q, st);
}
-_cl_kernel::sampler_argument::sampler_argument() :
- argument(0) {
-}
-
void
_cl_kernel::sampler_argument::set(size_t size, const void *value) {
if (size != sizeof(cl_sampler))
@@ -392,7 +375,8 @@
}
void
-_cl_kernel::sampler_argument::bind(exec_context &ctx) {
+_cl_kernel::sampler_argument::bind(exec_context &ctx,
+ const clover::module::argument &marg) {
size_t idx = ctx.samplers.size();
ctx.samplers.resize(idx + 1);
diff --git a/src/gallium/state_trackers/clover/core/kernel.hpp b/src/gallium/state_trackers/clover/core/kernel.hpp
index fd64f3e..e508e3c 100644
--- a/src/gallium/state_trackers/clover/core/kernel.hpp
+++ b/src/gallium/state_trackers/clover/core/kernel.hpp
@@ -68,14 +68,11 @@
public:
class argument {
public:
- argument(size_t size);
+ argument();
/// \a true if the argument has been set.
bool set() const;
- /// Argument size in the input buffer.
- size_t size() const;
-
/// Storage space required for the referenced object.
virtual size_t storage() const;
@@ -84,19 +81,19 @@
/// Allocate the necessary resources to bind the specified
/// object to this argument, and update \a ctx accordingly.
- virtual void bind(exec_context &ctx) = 0;
+ virtual void bind(exec_context &ctx,
+ const clover::module::argument &marg) = 0;
/// Free any resources that were allocated in bind().
virtual void unbind(exec_context &ctx) = 0;
protected:
- size_t __size;
bool __set;
};
_cl_kernel(clover::program &prog,
const std::string &name,
- const std::vector<clover::module::argument> &args);
+ const std::vector<clover::module::argument> &margs);
void launch(clover::command_queue &q,
const std::vector<size_t> &grid_offset,
@@ -122,19 +119,20 @@
scalar_argument(size_t size);
virtual void set(size_t size, const void *value);
- virtual void bind(exec_context &ctx);
+ virtual void bind(exec_context &ctx,
+ const clover::module::argument &marg);
virtual void unbind(exec_context &ctx);
private:
+ size_t size;
std::vector<uint8_t> v;
};
class global_argument : public argument {
public:
- global_argument(size_t size);
-
virtual void set(size_t size, const void *value);
- virtual void bind(exec_context &ctx);
+ virtual void bind(exec_context &ctx,
+ const clover::module::argument &marg);
virtual void unbind(exec_context &ctx);
private:
@@ -143,12 +141,11 @@
class local_argument : public argument {
public:
- local_argument();
-
virtual size_t storage() const;
virtual void set(size_t size, const void *value);
- virtual void bind(exec_context &ctx);
+ virtual void bind(exec_context &ctx,
+ const clover::module::argument &marg);
virtual void unbind(exec_context &ctx);
private:
@@ -157,10 +154,9 @@
class constant_argument : public argument {
public:
- constant_argument();
-
virtual void set(size_t size, const void *value);
- virtual void bind(exec_context &ctx);
+ virtual void bind(exec_context &ctx,
+ const clover::module::argument &marg);
virtual void unbind(exec_context &ctx);
private:
@@ -170,10 +166,9 @@
class image_rd_argument : public argument {
public:
- image_rd_argument();
-
virtual void set(size_t size, const void *value);
- virtual void bind(exec_context &ctx);
+ virtual void bind(exec_context &ctx,
+ const clover::module::argument &marg);
virtual void unbind(exec_context &ctx);
private:
@@ -183,10 +178,9 @@
class image_wr_argument : public argument {
public:
- image_wr_argument();
-
virtual void set(size_t size, const void *value);
- virtual void bind(exec_context &ctx);
+ virtual void bind(exec_context &ctx,
+ const clover::module::argument &marg);
virtual void unbind(exec_context &ctx);
private:
@@ -196,10 +190,9 @@
class sampler_argument : public argument {
public:
- sampler_argument();
-
virtual void set(size_t size, const void *value);
- virtual void bind(exec_context &ctx);
+ virtual void bind(exec_context &ctx,
+ const clover::module::argument &marg);
virtual void unbind(exec_context &ctx);
private: