Избегайте копий массивов в MEX-функциях

MEX-функции часто могут улучшать производительность путем управления, когда функция делает копии больших массивов данных. Можно избежать ненужных копий массивов в этих случаях:

  • MEX-функция получает доступ к входному массиву, но элементы массива не изменяются.

  • MEX-функция изменяет входной массив, и измененный массив возвращен в функцию вызова.

Входной массив, не измененный

Это MEX-функции суммируют элементы входного массива, но не изменяют массив. Присвоение inputs[0] к переменной const matlab::data::TypedArray<double> позволяет вам использовать основанный на области значений цикл for, чтобы суммировать элементы. При определении массива, когда const гарантирует, что переменная inArray остается разделяемой с входным массивом.

#include "mex.hpp"
#include "mexAdapter.hpp"

using namespace matlab::data;
using matlab::mex::ArgumentList;

class MexFunction : public matlab::mex::Function {
    ArrayFactory factory;
public:
    void operator()(ArgumentList outputs, ArgumentList inputs) {
    double sm = 0;
    const TypedArray<double> inArray = inputs[0];
    for (auto& elem : inArray) {
        sm += elem;
    }
    outputs[0] = factory.createScalar(sm);
    }
};

После сохранения этого кода в файле (названный addArrayElements.cpp в этом примере), скомпилируйте его с функцией mex. Вызовите эту MEX-функцию от MATLAB® с двойным массивом как входной параметр.

mex addArrayElements.cpp
b = addArrayElements([1:1e7]);
b =

        5.0000e+13

Измененный входной массив

Эта MEX-функция заменяет отрицательные величины во входном массиве с нулями. Эта операция изменяет входной массив, таким образом, вход не может остаться разделяемым с измененным массивом. После проверки входного массива функция не использует inputs[0] снова. Поэтому подтвержденный массив inputs[0] перемещен в переменную matlab::data::TypedArray<double>, чтобы позволить использованию основанного на области значений цикла for изменить элементы массива.

Чтобы предотвратить копию входного массива, переместите input[0] в matlab::data::TypedArray<double> с помощью std::move(), который подкачивает память, сопоставленную с входным массивом к переменной largeArray.

#include "mex.hpp"
#include "mexAdapter.hpp"

using namespace matlab::data;
using matlab::mex::ArgumentList;

class MexFunction : public matlab::mex::Function {
public:
    void operator()(ArgumentList outputs, ArgumentList inputs) {
        checkArguments(inputs);
        TypedArray<double> largeArray = std::move(inputs[0]);
        for (auto& elem : largeArray) {
            if (elem < 0) {
                elem = 0;
            }
        }
        outputs[0] = largeArray;
    }

    void checkArguments(ArgumentList inputs) {
        std::shared_ptr<matlab::engine::MATLABEngine> matlabPtr = getEngine();
        ArrayFactory factory;

        if (inputs[0].getType() != ArrayType::DOUBLE ||
            inputs[0].getType() == ArrayType::COMPLEX_DOUBLE) {
            matlabPtr->feval(u"error", 0, 
                std::vector<Array>({ factory.createScalar("Incorrect input") }));
        }
    }
};

После сохранения этого кода в файле (названный removeNegativeNumbers.cpp в этом примере), скомпилируйте его с функцией mex.

mex removeNegativeNumbers.cpp

Вызовите эту MEX-функцию от функции MATLAB. Повторно присвойте измененный массив той же переменной.

function arry = processArray(arry)
    arry = removeNegativeNumbers(arry);
    ....
end
    

Например, вызовите функцию processArray с большим массивом, возвращенным функцией randn MATLAB.

A = processArray(randn(10000));
min(A(:))

ans =

     0

Похожие темы

Для просмотра документации необходимо авторизоваться на сайте