def _to_tensor(gdf: cudf.DataFrame, dtype, to_cpu=False): if gdf.empty: return g = gdf.to_dlpack() cols = gdf.columns t = from_dlpack(g).type(dtype) del g, gdf return t, cols