works with new cutorch