提交 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:: ...@@ -163,7 +163,7 @@ would be::
ls = 1; ls = 1;
gs = 256; 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): ...@@ -329,7 +329,7 @@ class GpuKernelBase(object):
setargs = '\n '.join(setargs) setargs = '\n '.join(setargs)
return """ 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}) {{ {args}) {{
{setargs} {setargs}
...@@ -419,7 +419,7 @@ int {fname}(unsigned int nd, size_t *ldim, size_t *gdim, size_t shared, ...@@ -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. 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): def forward_string_meth(name):
...@@ -1393,7 +1393,7 @@ KERNEL void eye(GLOBAL_MEM %(ctype)s *a, ga_size n, ga_size m) { ...@@ -1393,7 +1393,7 @@ KERNEL void eye(GLOBAL_MEM %(ctype)s *a, ga_size n, ga_size m) {
ls = 1; ls = 1;
gs = 256; 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) { if (err != GA_NO_ERROR) {
PyErr_Format(PyExc_RuntimeError, PyErr_Format(PyExc_RuntimeError,
"gpuarray error: kEye: %%s. n%%lu, m=%%lu.", "gpuarray error: kEye: %%s. n%%lu, m=%%lu.",
......
...@@ -37,7 +37,7 @@ int APPLY_SPECIFIC(tstgpueye)(PyArrayObject *n, PyArrayObject *m, ...@@ -37,7 +37,7 @@ int APPLY_SPECIFIC(tstgpueye)(PyArrayObject *n, PyArrayObject *m,
ls = 1; ls = 1;
gs = 256; gs = 256;
/* The eye_call name comes from the kernel declaration above. */ /* 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) { if (err != GA_NO_ERROR) {
PyErr_Format(PyExc_RuntimeError, PyErr_Format(PyExc_RuntimeError,
"gpuarray error: kEye: %s. n%lu, m=%lu.", "gpuarray error: kEye: %s. n%lu, m=%lu.",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论