Update README.md

Author: Rabab53

parts/gpu/README.md

@@ -37,7 +37,7 @@ The notebook presents a series of benchmarks comparing execution times across di
    - Uses `KernelAbstractions.jl` to write code that can run on both CPU and GPU.
    - Offers flexibility with performance close to the `CUDA.jl` implementation.
 
-Here's the plot showing the benchmark results on a single A100 GPU in Perlmutter:
+Here's the plot showing the benchmark results on a single A100 GPU in Perlmutter using problem of size `10240x10240` and number of time steps `2000`:
  <img src="img/benchmarking.png" alt="Benchmark Results" width="500">