line 41 GradientScalarOutput and GradientVecOutput shouldn't be templated on device
GradientScalarOutput
GradientVecOutput