Fix compile without cuda