微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

c – MKL或BLAS例程将矢量乘以标量不合适的位置

我在模拟软件中工作,在阵列上完成的许多操作之一是按数字缩放矢量.

我有这样的代码

//Just some initialization code,don't bother about this part
int n = 10000;
std::vector<double> input(n,42.0);
std::vector<double> output(input.size());

double alpha = 69.0;

//the actual calculation:
for (size_t i = 0; i < n; ++i) {
    output[i] = input[i] * alpha;
}

我有MKL库,所以如果我的计算是“就地”完成的,可以写下面的内容

cblas_dscal(n,alpha,&input[0],1);

但是,这会改变输入变量,这不是我想要的.

我尝试使用mkl_domatcopy(),但这个操作非常慢.

解决方法

我想出的解决方案是调用cblas_dcopy()然后调用cblas_dscal().

它不是所有世界中最好的,但它仍然比原始循环更快.

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐