Open gmarkall opened 4 days ago
Something like this works today:
from numba import cuda
@cuda.jit
def kernel():
for g, d in zip((4, 1, 1), (256, 1, 1)):
print(g * d)
kernel[1, 1]()
But it won't work with Dim3s, because they're not iterable.
E.g. It would be convenient to be able to do things like
which is not presently possible with the current implementation of
Dim3
.cc @brycelelbach