Add group feature to ndarray
Single GPU works with new interface