Fix cuda incomplete type issue