Skip to content

Commit 80ee258

Browse files
committed
core: allow template specialization according to the scalar type
1 parent f4dbd90 commit 80ee258

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

include/eigenpy/details.hpp

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,11 +19,12 @@
1919
namespace eigenpy {
2020

2121
template <typename EigenType,
22-
typename BaseType = typename get_eigen_base_type<EigenType>::type>
22+
typename BaseType = typename get_eigen_base_type<EigenType>::type,
23+
typename Scalar = typename EigenType::Scalar>
2324
struct expose_eigen_type_impl;
2425

25-
template <typename MatType>
26-
struct expose_eigen_type_impl<MatType, Eigen::MatrixBase<MatType> > {
26+
template <typename MatType, typename Scalar>
27+
struct expose_eigen_type_impl<MatType, Eigen::MatrixBase<MatType>, Scalar> {
2728
static void run() {
2829
if (check_registration<MatType>()) return;
2930

@@ -40,8 +41,9 @@ struct expose_eigen_type_impl<MatType, Eigen::MatrixBase<MatType> > {
4041
};
4142

4243
#ifdef EIGENPY_WITH_TENSOR_SUPPORT
43-
template <typename TensorType>
44-
struct expose_eigen_type_impl<TensorType, Eigen::TensorBase<TensorType> > {
44+
template <typename TensorType, typename Scalar>
45+
struct expose_eigen_type_impl<TensorType, Eigen::TensorBase<TensorType>,
46+
Scalar> {
4547
static void run() {
4648
if (check_registration<TensorType>()) return;
4749

0 commit comments

Comments
 (0)