fix build without cuda