It appears the issue lies in:
AT_DISPATCH_FLOATING_TYPES(
      data_col.type(), ...

The correct way to do this now is to use .scalar_type(), so it should read:
AT_DISPATCH_FLOATING_TYPES(
      data_col.scalar_type(), ...