diff --git a/example/heatEquation2D/src/heatEquation2D.cpp b/example/heatEquation2D/src/heatEquation2D.cpp index 46a9ad9620f..d356af0e6f1 100644 --- 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 numNodes{128, 128}; + constexpr alpaka::Vec numNodes{256, 256}; constexpr alpaka::Vec haloSize{2, 2}; constexpr alpaka::Vec 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(extent[1] - 1); @@ -100,8 +100,8 @@ auto example(TAccTag const&) -> int constexpr alpaka::Vec 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 chunkSize{ySize, xSize}; constexpr auto sharedMemSize = (ySize + halo) * (xSize + halo);