Fix warning when compiling in CUDA aware code.