Improve Sensitivity Speed.

This commit is contained in:
kradchen
2023-05-22 14:06:29 +08:00
parent d0c95811a7
commit a4a972f99f

View File

@@ -90,6 +90,7 @@ TEST_F(Sensitivity_Test, combineSensitivity){
EXPECT_EQ(output.size(), sens.size()); EXPECT_EQ(output.size(), sens.size());
EXPECT_EQ(output[0].getDataSize(), sens[0].getDataSize()); EXPECT_EQ(output[0].getDataSize(), sens[0].getDataSize());
#pragma omp parallel for
for (size_t i = 0; i < output.size(); i++) for (size_t i = 0; i < output.size(); i++)
{ {
for (size_t j = 0; j < output[i].getDataSize(); j++) for (size_t j = 0; j < output[i].getDataSize(); j++)
@@ -120,12 +121,12 @@ TEST_F(Sensitivity_Test,precalcSensitivity){
auto sens = m2.read4d("result"); auto sens = m2.read4d("result");
EXPECT_EQ(output.size(), sens.size()); EXPECT_EQ(output.size(), sens.size());
EXPECT_EQ(output[0].getDataSize(), sens[0].getDataSize()); EXPECT_EQ(output[0].getDataSize(), sens[0].getDataSize());
#pragma omp parallel for
for (size_t i = 0; i < output.size(); i++) for (size_t i = 0; i < output.size(); i++)
{ {
for (size_t j = 0; j < output[i].getDataSize(); j++) for (size_t j = 0; j < output[i].getDataSize(); j++)
{ {
ASSERT_DOUBLE_AE(output[i].getData()[j],sens[i].getData()[j])<<"Matrix index:"<<i<<", value index:"<<j; EXPECT_DOUBLE_AE(output[i].getData()[j],sens[i].getData()[j])<<"Matrix index:"<<i<<", value index:"<<j;
} }
} }