add missing #include <cuda_runtime.h>