Extend support for diagonal matrices on CUDA devices

fwyzard · smuzaffar · commit 017ce22865bc · 2021-07-15T19:35:23.000+02:00
diff --git a/Eigen/src/Core/DiagonalMatrix.h b/Eigen/src/Core/DiagonalMatrix.h
@@ -366,6 +366,7 @@ template<> struct AssignmentKind<DenseShape,DiagonalShape> { typedef Diagonal2De
 template< typename DstXprType, typename SrcXprType, typename Functor>
 struct Assignment<DstXprType, SrcXprType, Functor, Diagonal2Dense>
 {
+  EIGEN_DEVICE_FUNC
   static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar> &/*func*/)
   {
     Index dstRows = src.rows();
@@ -377,9 +378,11 @@ struct Assignment<DstXprType, SrcXprType, Functor, Diagonal2Dense>
     dst.diagonal() = src.diagonal();
   }
   
+  EIGEN_DEVICE_FUNC
   static void run(DstXprType &dst, const SrcXprType &src, const internal::add_assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar> &/*func*/)
   { dst.diagonal() += src.diagonal(); }
   
+  EIGEN_DEVICE_FUNC
   static void run(DstXprType &dst, const SrcXprType &src, const internal::sub_assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar> &/*func*/)
   { dst.diagonal() -= src.diagonal(); }
 };

Original file line number	Diff line number	Diff line change
`@@ -366,6 +366,7 @@ template<> struct AssignmentKind<DenseShape,DiagonalShape> { typedef Diagonal2De`
`366`	`366`	`template< typename DstXprType, typename SrcXprType, typename Functor>`
`367`	`367`	`struct Assignment<DstXprType, SrcXprType, Functor, Diagonal2Dense>`
`368`	`368`	`{`
	`369`	`+ EIGEN_DEVICE_FUNC`
`369`	`370`	`static void run(DstXprType &dst, const SrcXprType &src, const internal::assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar> &/func/)`
`370`	`371`	`{`
`371`	`372`	`Index dstRows = src.rows();`
`@@ -377,9 +378,11 @@ struct Assignment<DstXprType, SrcXprType, Functor, Diagonal2Dense>`
`377`	`378`	`dst.diagonal() = src.diagonal();`
`378`	`379`	`}`
`379`	`380`
	`381`	`+ EIGEN_DEVICE_FUNC`
`380`	`382`	`static void run(DstXprType &dst, const SrcXprType &src, const internal::add_assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar> &/func/)`
`381`	`383`	`{ dst.diagonal() += src.diagonal(); }`
`382`	`384`
	`385`	`+ EIGEN_DEVICE_FUNC`
`383`	`386`	`static void run(DstXprType &dst, const SrcXprType &src, const internal::sub_assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar> &/func/)`
`384`	`387`	`{ dst.diagonal() -= src.diagonal(); }`
`385`	`388`	`};`