本文整理汇总了C++中dmatrix::set方法的典型用法代码示例。如果您正苦于以下问题：C++ dmatrix::set方法的具体用法？C++ dmatrix::set怎么用？C++ dmatrix::set使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类dmatrix的用法示例。
在下文中一共展示了dmatrix::set方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的C++代码示例。
示例1: orthogonalize
//.........这里部分代码省略.........
#endif
}
}
/* CPU version */
if (pu == CPU) {
/* multiplication by triangular matrix */
for (auto& e: wfs__) {
/* wave functions are complex, transformation matrix is complex */
if (std::is_same<T, double_complex>::value) {
linalg<CPU>::trmm('R', 'U', 'N', e->pw_coeffs().num_rows_loc(), n__, double_complex(1, 0),
reinterpret_cast<double_complex*>(o__.template at<CPU>()), o__.ld(),
e->pw_coeffs().prime().at<CPU>(0, N__), e->pw_coeffs().prime().ld());
if (e->has_mt() && e->mt_coeffs().num_rows_loc()) {
linalg<CPU>::trmm('R', 'U', 'N', e->mt_coeffs().num_rows_loc(), n__, double_complex(1, 0),
reinterpret_cast<double_complex*>(o__.template at<CPU>()), o__.ld(),
e->mt_coeffs().prime().at<CPU>(0, N__), e->mt_coeffs().prime().ld());
}
}
/* wave functions are real (psi(G) = psi^{*}(-G)), transformation matrix is real */
if (std::is_same<T, double>::value) {
linalg<CPU>::trmm('R', 'U', 'N', 2 * e->pw_coeffs().num_rows_loc(), n__, 1.0,
reinterpret_cast<double*>(o__.template at<CPU>()), o__.ld(),
reinterpret_cast<double*>(e->pw_coeffs().prime().at<CPU>(0, N__)), 2 * e->pw_coeffs().prime().ld());
if (e->has_mt() && e->mt_coeffs().num_rows_loc()) {
linalg<CPU>::trmm('R', 'U', 'N', 2 * e->mt_coeffs().num_rows_loc(), n__, 1.0,
reinterpret_cast<double*>(o__.template at<CPU>()), o__.ld(),
reinterpret_cast<double*>(e->mt_coeffs().prime().at<CPU>(0, N__)), 2 * e->mt_coeffs().prime().ld());
}
}
}
}
#ifdef __GPU
if (pu == GPU) {
/* multiplication by triangular matrix */
for (auto& e: wfs__) {
if (std::is_same<T, double_complex>::value) {
double_complex alpha(1, 0);
linalg<GPU>::trmm('R', 'U', 'N', e->pw_coeffs().num_rows_loc(), n__, &alpha,
reinterpret_cast<double_complex*>(o__.template at<GPU>()), o__.ld(),
e->pw_coeffs().prime().at<GPU>(0, N__), e->pw_coeffs().prime().ld());
if (e->has_mt() && e->mt_coeffs().num_rows_loc()) {
linalg<GPU>::trmm('R', 'U', 'N', e->mt_coeffs().num_rows_loc(), n__, &alpha,
reinterpret_cast<double_complex*>(o__.template at<GPU>()), o__.ld(),
e->mt_coeffs().prime().at<GPU>(0, N__), e->mt_coeffs().prime().ld());
}
/* alpha should not go out of the scope, so wait */
acc::sync_stream(-1);
}
if (std::is_same<T, double>::value) {
double alpha{1};
linalg<GPU>::trmm('R', 'U', 'N', 2 * e->pw_coeffs().num_rows_loc(), n__, &alpha,
reinterpret_cast<double*>(o__.template at<GPU>()), o__.ld(),
reinterpret_cast<double*>(e->pw_coeffs().prime().at<GPU>(0, N__)), 2 * e->pw_coeffs().prime().ld());
if (e->has_mt() && e->mt_coeffs().num_rows_loc()) {
linalg<GPU>::trmm('R', 'U', 'N', 2 * e->mt_coeffs().num_rows_loc(), n__, &alpha,
reinterpret_cast<double*>(o__.template at<GPU>()), o__.ld(),
reinterpret_cast<double*>(e->mt_coeffs().prime().at<GPU>(0, N__)), 2 * e->mt_coeffs().prime().ld());
}
acc::sync_stream(-1);
}
}
acc::sync_stream(-1);
}
#endif
} else { /* parallel transformation */
sddk::timer t1("sddk::wave_functions::orthogonalize|potrf");
if (int info = linalg<CPU>::potrf(n__, o__)) {
std::stringstream s;
s << "error in factorization, info = " << info;
TERMINATE(s);
}
t1.stop();
sddk::timer t2("sddk::wave_functions::orthogonalize|trtri");
if (linalg<CPU>::trtri(n__, o__)) {
TERMINATE("error in inversion");
}
t2.stop();
/* o is upper triangular matrix */
for (int i = 0; i < n__; i++) {
for (int j = i + 1; j < n__; j++) {
o__.set(j, i, 0);
}
}
/* phi is transformed into phi, so we can't use it as the output buffer; use tmp instead and then overwrite phi */
for (auto& e: wfs__) {
transform(pu, *e, N__, n__, o__, 0, 0, tmp__, 0, n__);
e->copy_from(tmp__, 0, n__, N__, pu);
}
}
}