fix: add option for cuda