• brkirch's avatar
    Fix cumsum for MPS in newer torch · a255dac4
    brkirch authored
    The prior fix assumed that testing int16 was enough to determine if a fix is needed, but a recent fix for cumsum has int16 working but not bool.
    a255dac4
devices.py 5.34 KB