提交 e4681a78 authored 作者: Shawn Tan's avatar Shawn Tan

Moved error check into bounds check, and modified cache version.

上级 7121c2fb
...@@ -1662,14 +1662,13 @@ KERNEL void eye(GLOBAL_MEM %(ctype)s *a, ga_size n, ga_size m, ga_ssize k) { ...@@ -1662,14 +1662,13 @@ KERNEL void eye(GLOBAL_MEM %(ctype)s *a, ga_size n, ga_size m, ga_ssize k) {
row_off = (size_t) (k < 0?-k:0); row_off = (size_t) (k < 0?-k:0);
if (row_off < dims[0] && col_off < dims[1]) { if (row_off < dims[0] && col_off < dims[1]) {
err = eye_call(1, &gs, &ls, 0, %(z)s->ga.data, dims[0], dims[1], k); err = eye_call(1, &gs, &ls, 0, %(z)s->ga.data, dims[0], dims[1], k);
} if (err != GA_NO_ERROR) {
PyErr_Format(PyExc_RuntimeError,
if (err != GA_NO_ERROR) { "gpuarray error: kEye: %%s. n%%lu, m=%%lu.",
PyErr_Format(PyExc_RuntimeError, GpuKernel_error(&%(kname)s, err),
"gpuarray error: kEye: %%s. n%%lu, m=%%lu.", (unsigned long)dims[0], (unsigned long)dims[1]);
GpuKernel_error(&%(kname)s, err), %(fail)s;
(unsigned long)dims[0], (unsigned long)dims[1]); }
%(fail)s;
} }
if(%(sync)d) if(%(sync)d)
...@@ -1679,5 +1678,4 @@ KERNEL void eye(GLOBAL_MEM %(ctype)s *a, ga_size n, ga_size m, ga_ssize k) { ...@@ -1679,5 +1678,4 @@ KERNEL void eye(GLOBAL_MEM %(ctype)s *a, ga_size n, ga_size m, ga_ssize k) {
return s return s
def c_code_cache_version(self): def c_code_cache_version(self):
# return (7,) return (7,)
return
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论