消除中间特征数组

Eliminate Intermediate Eigen Arrays

本文关键字：数组 特征 中间      更新时间：2023-10-16

特征是创建任何中间数组来计算x还是只将值放入simd寄存器并进行计算？

一般来说，如何知道本征制造了多少中间体？

Eigen 会在循环的每个周期中为中间体分配新的内存吗？

有没有办法确保本征不会做任何中间体？它有像"EIGEN_NO_INTERMEDIATE"这样的宏吗？

#include <Eigen/Eigen>
#include <iostream>
using namespace Eigen;
template<typename T>
void fill(T& x) {
for (int i = 0; i < x.size(); ++i) x.data()[i] = i + 1;
}
int main() {
int n = 10;  // n is actually about 400
ArrayXXf x(n, n);
ArrayXf y(n);
fill(x);
fill(y);
for (int i = 0; i < 10; ++i) {  // many cycles
x = x * ((x.colwise() / y).rowwise() / y.transpose()).exp();
}
std::cout << x << "n";
}

您可以在DenseStorage构造函数中添加一个钩子，如下所示：

#include <iostream>
static long int nb_temporaries;
inline void on_temporary_creation(long int size) {
if(size!=0) nb_temporaries++;
}
// must be defined before including any Eigen header!
#define EIGEN_DENSE_STORAGE_CTOR_PLUGIN { on_temporary_creation(size); }
#define VERIFY_EVALUATION_COUNT(XPR,N) {
nb_temporaries = 0; 
XPR; 
if(nb_temporaries!=N) { std::cerr << "nb_temporaries == " << nb_temporaries << "n"; }
}
#include <Eigen/Core>
using namespace Eigen;
template<typename T>
void fill(T& x) { for(int i=0; i<x.size(); ++i) x(i)= i+1; }
int main() {
int n=10;
ArrayXXf x(n,n); fill(x);
ArrayXf y(n); fill(y);
for(int i=0; i<10; ++i)
{
VERIFY_EVALUATION_COUNT( x = x * ((x.colwise()/y).rowwise()/y.transpose()).exp(), 0);
}
std::cout << x << 'n';
}

从本质上讲，这就是 Eigen 在某些时候在其测试套件中所做的： 有关测试套件中的原始定义，请参阅此处，有关测试套件中的示例用法，请参阅此处。

或者，如果您只关心中间内存分配，则可以尝试宏EIGEN_RUNTIME_NO_MALLOC- 这将允许固定大小的表达式计算为临时表达式，因为它们只会在堆栈上分配。

相关文章：