API: Fixed single-element indexing
This commit is contained in:
@@ -71,7 +71,7 @@ public:
|
||||
//scalar
|
||||
if(a->shape()[0]==1 && a->shape()[1]==1)
|
||||
{
|
||||
kernel_.setSizeArg(current_arg_++, a->start()[0]);
|
||||
kernel_.setSizeArg(current_arg_++, a->start()[0] + a->start()[1]*a->ld());
|
||||
}
|
||||
//array
|
||||
else if(a->shape()[0]>1 && a->shape()[1]==1)
|
||||
|
Reference in New Issue
Block a user