Looks like fixed, need to check on cuda working machine