Skip to content
Snippets Groups Projects
Commit b16f4dc8 authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Parse "N" as a shape/stride spec in kernel input.

parent 8e4493cd
Branches
Tags
No related merge requests found
......@@ -128,13 +128,17 @@ class ArrayArg:
from pymbolic import parse
strides = parse(strides)
strides = tuple(strides)
if not isinstance(shape, tuple):
shape = (shape,)
if shape is not None:
if isinstance(shape, str):
from pymbolic import parse
shape = parse(shape)
if not isinstance(shape, tuple):
shape = (shape,)
from pyopencl.compyte.array import (
f_contiguous_strides,
c_contiguous_strides)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment