6

Boost.Proto を使用して、組み込みのドメイン固有言語を Eigen ライブラリで実装された一連の行列演算に変換したいと考えています。効率が重要なので、proto で Eigen 式テンプレートを生成し、時期尚早の評価を回避する必要があります。

行列の乗算式を生成できる簡単な文法を実装しました。以下のコードは警告なしでコンパイルされ (g++ 4.8.0 および Intel C++ 2013.3 で Boost 1.54.0 および Eigen 3.1.3 を使用)、私の式が 1 つの乗算演算しかない限り機能します。チェーンに乗算を追加するとすぐにクラッシュします。Valgrind は、評価が完了する前に Eigen::GeneralProduct 式テンプレートのテンポラリの 1 つが破棄されるためだと教えてくれました。

なぜこれが起こるのか、またはそれを防ぐために何ができるのかわかりません。すべての助けに感謝します!

#include <iostream>

#include <boost/fusion/container.hpp>
#include <boost/mpl/int.hpp>
#include <boost/mpl/void.hpp>
#include <boost/proto/proto.hpp>
#include <boost/ref.hpp>
#include <boost/type_traits/remove_const.hpp>
#include <boost/type_traits/remove_reference.hpp>
#include <boost/utility.hpp>

#include <Eigen/Dense>

namespace fusion = boost::fusion;
namespace mpl = boost::mpl;
namespace proto = boost::proto;

typedef Eigen::Matrix<float, Eigen::Dynamic, Eigen::Dynamic> matrix;

// Placeholders

const proto::terminal<mpl::int_<0> >::type I1 = {{}};
const proto::terminal<mpl::int_<1> >::type I2 = {{}};
const proto::terminal<mpl::int_<2> >::type I3 = {{}};

// Grammar

template<class Rule, class Callable = proto::callable>
struct External :
    proto::when<Rule, proto::external_transform> {};

struct matmul_transform : proto::callable {
    template<class Sig> struct result;

    template<class This, class MatrixExpr1, class MatrixExpr2>
    struct result<This(MatrixExpr1, MatrixExpr2)> {
            typedef typename Eigen::ProductReturnType<
                    typename boost::remove_const<typename boost::remove_reference<MatrixExpr1>::type>::type,
                    typename boost::remove_const<typename boost::remove_reference<MatrixExpr2>::type>::type>::Type
                    type;
    };

    template<class MatrixExpr1, class MatrixExpr2>
    typename result<matmul_transform(MatrixExpr1, MatrixExpr2)>::type
    operator()(const MatrixExpr1 &a, const MatrixExpr2 &b) const {
            return a * b;
    }
};


struct MatmulGrammar;

struct InputPlaceholder : proto::terminal<proto::_> {};

struct MatrixMultiplication :
    proto::multiplies<MatmulGrammar, MatmulGrammar> {};

struct MatmulGrammar : proto::or_<
    External<InputPlaceholder>,
    External<MatrixMultiplication> > {};

struct matmul_transforms : proto::external_transforms<
    proto::when<MatrixMultiplication, matmul_transform(MatmulGrammar(proto::_left), MatmulGrammar(proto::_right))>,
    proto::when<InputPlaceholder, proto::functional::at(proto::_data, proto::_value)> > {};

int main() {
    matrix mat1(2,2), mat2(2,2), mat3(2,2), result(2,2);

    mat1 << 1, 2, 3, 4;
    mat2 << 5, 6, 7, 8;
    mat3 << 1, 3, 6, 9;

    MatmulGrammar mmg;

    // THIS WORKS:
    result = mmg(I1 * I2,
            mpl::void_(),
            (proto::data = fusion::make_vector(boost::cref(mat1), boost::cref(mat2), boost::cref(mat3)),
             proto::transforms = matmul_transforms()));

    std::cout << result << std::endl;

    // THIS CRASHES:
    result = mmg(I1 * I2 * I3,
            mpl::void_(),
            (proto::data = fusion::make_vector(boost::cref(mat1), boost::cref(mat2), boost::cref(mat3)),
             proto::transforms = matmul_transforms()));

    std::cout << result << std::endl;

    return 0;
}
4

2 に答える 2