From a4f3a8e9548151ca043f86399713c7b82f296e79 Mon Sep 17 00:00:00 2001
From: Giovanni La Mura <giovanni.lamura@inaf.it>
Date: Tue, 10 Jun 2025 17:33:34 +0200
Subject: [PATCH] Get wk_value index from loop indices
---
src/trapping/cfrfme.cpp | 10 +---------
1 file changed, 1 insertion(+), 9 deletions(-)
diff --git a/src/trapping/cfrfme.cpp b/src/trapping/cfrfme.cpp
index 4c8010d..deba6ef 100644
--- a/src/trapping/cfrfme.cpp
+++ b/src/trapping/cfrfme.cpp
@@ -395,22 +395,15 @@ void frfme(string data_file, string output_path) {
for (int j80 = jlmf; j80 <= jlml; j80++) {
dcomplex *vec_w = new dcomplex[nkv * nkv]();
dcomplex **w = new dcomplex*[nkv];
- // dcomplex *wk_local = new dcomplex[nlmmt]();
for (int wi = 0; wi < nkv; wi++) w[wi] = vec_w + wi * nkv;
dcomplex wk_value;
- int wk_index = 0;
- int nkvs = nkv * nkv;
for (int jy50 = 0; jy50 < nkv; jy50++) {
for (int jx50 = 0; jx50 < nkv; jx50++) {
- // for (int wi = 0; wi < nlmmt; wi++) wk_local[wi] = tt1->wk[wk_index++];
- // w[jx50][jy50] = wk_local[j80 - 1];
- wk_value = tt1->wk[wk_index + j80 - 1];
- wk_index += nlmmt;
+ wk_value = tt1->wk[jy50 * nkv * nlmmt + jx50 * nlmmt + j80 - 1];
w[jx50][jy50] = wk_value;
} // jx50
} // jy50 loop
int ixyz = 0;
- for (int wj = 0; wj < nrvc; wj++) tfrfme->wsum[j80 - 1][wj] = cc0;
for (int iz75 = 0; iz75 < nzv; iz75++) {
double z = _zv[iz75] + frsh;
for (int iy70 = 0; iy70 < nyv; iy70++) {
@@ -442,7 +435,6 @@ void frfme(string data_file, string output_path) {
} // iz75 loop
delete[] vec_w;
delete[] w;
- // delete[] wk_local;
} // j80 loop
#ifdef USE_NVTX
nvtxRangePop();
--
GitLab