https://stackoverflow.com/questions/64684876/numba-cuda-changing-xr-dataarray-shape-causes-call-to-cumemcpydtoh-results-in