提交 5c9ad24e authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Change the order of ls and gs in the wrapper.

上级 b3af29cc
......@@ -163,7 +163,7 @@ would be::
ls = 1;
gs = 256;
err = k_call(1, &ls, &gs, 0, input->ga.data, dims[0], dims[1]);
err = k_call(1, &gs, &ls, 0, input->ga.data, dims[0], dims[1]);
// ...
......
......@@ -329,7 +329,7 @@ class GpuKernelBase(object):
setargs = '\n '.join(setargs)
return """
int {fname}(unsigned int nd, size_t *ldim, size_t *gdim, size_t shared,
int {fname}(unsigned int nd, size_t *gdim, size_t *ldim, size_t shared,
{args}) {{
{setargs}
......@@ -419,7 +419,7 @@ int {fname}(unsigned int nd, size_t *ldim, size_t *gdim, size_t shared,
The node that we need the cache version for.
"""
return (5, self.get_params(node).bin_id)
return (6, self.get_params(node).bin_id)
def forward_string_meth(name):
......@@ -1393,7 +1393,7 @@ KERNEL void eye(GLOBAL_MEM %(ctype)s *a, ga_size n, ga_size m) {
ls = 1;
gs = 256;
err = eye_call(1, &ls, &gs, 0, %(z)s->ga.data, dims[0], dims[1]);
err = eye_call(1, &gs, &ls, 0, %(z)s->ga.data, dims[0], dims[1]);
if (err != GA_NO_ERROR) {
PyErr_Format(PyExc_RuntimeError,
"gpuarray error: kEye: %%s. n%%lu, m=%%lu.",
......
......@@ -37,7 +37,7 @@ int APPLY_SPECIFIC(tstgpueye)(PyArrayObject *n, PyArrayObject *m,
ls = 1;
gs = 256;
/* The eye_call name comes from the kernel declaration above. */
err = eye_call(1, &ls, &gs, 0, (*z)->ga.data, dims[0], dims[1]);
err = eye_call(1, &gs, &ls, 0, (*z)->ga.data, dims[0], dims[1]);
if (err != GA_NO_ERROR) {
PyErr_Format(PyExc_RuntimeError,
"gpuarray error: kEye: %s. n%lu, m=%lu.",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论