fix of fix

trilinos · Sep 26, 2024 · 59359a4 · 59359a4
1 parent d5a9460
commit 59359a4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/packages/intrepid2/unit-test/Discretization/Basis/HCURL_HEX_In_FEM/test_02.hpp b/packages/intrepid2/unit-test/Discretization/Basis/HCURL_HEX_In_FEM/test_02.hpp
@@ -97,7 +97,7 @@ namespace Intrepid2 {
               ordinal_type team_size = teamPolicy.team_size_recommended(functor, Kokkos::ParallelForTag());
               *outStream << "Max Recommended team size: " << team_size << ", Requested team size: " << numPoints <<std::endl;
               team_size = std::min(team_size, numPoints);
-              teamPolicy = Kokkos::TeamPolicy<typename DeviceType::execution_space>(numCells, team_size,vectorSize);
+              teamPolicy = Kokkos::TeamPolicy<typename DeviceType::execution_space>(ncells, team_size,vectorSize);
 
               //Get the required size of the scratch space per team and per thread.
               int perThreadSpaceSize(0), perTeamSpaceSize(0);
@@ -117,7 +117,7 @@ namespace Intrepid2 {
               ordinal_type team_size = teamPolicy.team_size_recommended(functor, Kokkos::ParallelForTag());
               *outStream << "Max Recommended team size: " << team_size << ", Requested team size: " << numPoints <<std::endl;
               team_size = std::min(team_size, numPoints);
-              teamPolicy = Kokkos::TeamPolicy<typename DeviceType::execution_space>(numCells, team_size,vectorSize);         
+              teamPolicy = Kokkos::TeamPolicy<typename DeviceType::execution_space>(ncells, team_size,vectorSize);         
 
               //Get the required size of the scratch space per team and per thread.
               int perThreadSpaceSize(0), perTeamSpaceSize(0);

diff --git a/packages/intrepid2/unit-test/Discretization/Basis/HCURL_TET_In_FEM/test_02.hpp b/packages/intrepid2/unit-test/Discretization/Basis/HCURL_TET_In_FEM/test_02.hpp
@@ -97,7 +97,7 @@ namespace Intrepid2 {
               ordinal_type team_size = teamPolicy.team_size_recommended(functor, Kokkos::ParallelForTag());
               *outStream << "Max Recommended team size: " << team_size << ", Requested team size: " << numPoints <<std::endl;
               team_size = std::min(team_size, numPoints);
-              teamPolicy = Kokkos::TeamPolicy<typename DeviceType::execution_space>(numCells, team_size,vectorSize);
+              teamPolicy = Kokkos::TeamPolicy<typename DeviceType::execution_space>(ncells, team_size,vectorSize);
 
               //Get the required size of the scratch space per team and per thread.
               int perThreadSpaceSize(0), perTeamSpaceSize(0);
@@ -117,7 +117,7 @@ namespace Intrepid2 {
               ordinal_type team_size = teamPolicy.team_size_recommended(functor, Kokkos::ParallelForTag());
               *outStream << "Max Recommended team size: " << team_size << ", Requested team size: " << numPoints <<std::endl;
               team_size = std::min(team_size, numPoints);
-              teamPolicy = Kokkos::TeamPolicy<typename DeviceType::execution_space>(numCells, team_size,vectorSize);        
+              teamPolicy = Kokkos::TeamPolicy<typename DeviceType::execution_space>(ncells, team_size,vectorSize);        
 
               //Get the required size of the scratch space per team and per thread.
               int perThreadSpaceSize(0), perTeamSpaceSize(0);