script_v2.c

#include <stdio.h>
#include <stdlib.h>
#include <omp.h>


// Definir el tamaño de la cuadrícula y el número de pasos de simulación
#define SIZE 3000    // Tamaño de la cuadrícula (3000x3000)
#define STEPS 10000  // Número de iteraciones para la simulación
#define BLOCK_SIZE 64 // arbitrario

int main() {
    // Variables para medir tiempo
    double start, end, cpu_time_used;

    // Reservar memoria dinámica para la cuadrícula como un array unidimensional
    double *grid = malloc(SIZE * SIZE * sizeof(double));
    double *new_grid = malloc(SIZE * SIZE * sizeof(double));

    if (grid == NULL || new_grid == NULL) {
        // Verificar si la asignación de memoria falló
        fprintf(stderr, "Error: Unable to allocate memory for grid.\n");
        return EXIT_FAILURE;
    }

    start = omp_get_wtime(); // Comenzar medicion del tiempo

    int i, j, bi, bj, step; // Variables de control

    // Inicializar toda la cuadrícula con valores de 0.0
    for (i = 0; i < SIZE * SIZE; i++) {
        grid[i] = 0.0;
        new_grid[i] = 0.0;
    }

    // Establecer un valor inicial alto en el centro de la cuadrícula
    grid[(SIZE / 2) * SIZE + (SIZE / 2)] = 10000.0;

    // Inicio de la simulación de difusión
    for (step = 0; step < STEPS; step++) {
        /// Dividir la cuadrícula en bloques para optimizar el acceso a memoria
        #pragma omp parallel for collapse(2) private(i, j)
        for (bi = 1; bi < SIZE - 1; bi += BLOCK_SIZE) {
            for (bj = 1; bj < SIZE - 1; bj += BLOCK_SIZE) {
                for (i = bi; i < bi + BLOCK_SIZE && i < SIZE - 1; i++) {
                    for (j = bj; j < bj + BLOCK_SIZE && j < SIZE - 1; j++) {
                        new_grid[i * SIZE + j] = 0.25 * (
                            grid[(i - 1) * SIZE + j] +
                            grid[(i + 1) * SIZE + j] +
                            grid[i * SIZE + (j - 1)] +
                            grid[i * SIZE + (j + 1)]
                        );
                    }
                }
            }
        }

        // Intercambiar las cuadrículas
        double *temp = grid;
        grid = new_grid;
        new_grid = temp;
    }

    end = omp_get_wtime(); // Finalizar medicion del tiempo
    cpu_time_used = end - start;

    // Imprimir el valor final en el centro de la cuadrícula
    printf("Final grid center value: %f\n", grid[(SIZE / 2) * SIZE + (SIZE / 2)]);
    printf("Time: %f", cpu_time_used);

    // Liberar la memoria asignada para la cuadrícula
    free(grid);
    free(new_grid);

    return 0; // Terminar el programa con éxito
}