aligned with latest version of CUDAGraph