256 64 002

alpaka-group · Oct 23, 2024 · 9432c2e · 9432c2e
1 parent 36f0dac
commit 9432c2e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/example/heatEquation2D/src/heatEquation2D.cpp b/example/heatEquation2D/src/heatEquation2D.cpp
@@ -51,12 +51,12 @@ auto example(TAccTag const&) -> int
 
     // simulation defines
     // {Y, X}
-    constexpr alpaka::Vec<Dim, Idx> numNodes{128, 128};
+    constexpr alpaka::Vec<Dim, Idx> numNodes{256, 256};
     constexpr alpaka::Vec<Dim, Idx> haloSize{2, 2};
     constexpr alpaka::Vec<Dim, Idx> extent = numNodes + haloSize;
 
     constexpr uint32_t numTimeSteps = 6000;
-    constexpr double tMax = 0.04;
+    constexpr double tMax = 0.02;
 
     // x, y in [0, 1], t in [0, tMax]
     constexpr double dx = 1.0 / static_cast<double>(extent[1] - 1);
@@ -100,8 +100,8 @@ auto example(TAccTag const&) -> int
     constexpr alpaka::Vec<Dim, Idx> elemPerThread{1, 1};
 
     // Appropriate chunk size to split your problem for your Acc
-    constexpr Idx xSize = 32u;
-    constexpr Idx ySize = 32u;
+    constexpr Idx xSize = 64u;
+    constexpr Idx ySize = 64u;
     constexpr Idx halo = 2u;
     constexpr alpaka::Vec<Dim, Idx> chunkSize{ySize, xSize};
     constexpr auto sharedMemSize = (ySize + halo) * (xSize + halo);