Циклическое распределение столбцов матрицы ленточным способом по процессам

#include <stdio.h>
#include <mpi.h>
#include <stdlib.h>
 
const int ncolumns = 15;//количество строк
const int columnlen = 6;//количество столбцов
int main(int argc, char** argv)
{
    int** A = NULL;   // 2D массив, инициализированный в процессе 0 
    int* Adata = NULL;
    int* sendbufptr = NULL;
    int i, j;
    int** rows;  // 2D массив для строк 
    int* columndata;   // Хранилище данных для 1D массива строк 
 
    int my_rank, p;
    MPI_Datatype vector_mpi_t;
 
    MPI_Init(&argc, &argv);
 
    MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);
    MPI_Comm_size(MPI_COMM_WORLD, &p);
 
    // Инициализация 2D массива в процессе 0 и выделение памяти 
    if (my_rank == 0){
        A = (int**)malloc(ncolumns * sizeof(int*));
        Adata = (int*)malloc(ncolumns * columnlen * sizeof(int));
        for (i = 0; i < ncolumns; i++)
            A[i] = &(Adata[i * columnlen]);
        int y=1;
        for (i = 0; i < ncolumns; i++)
            for (j = 0; j < columnlen; j++)
                A[i][j] = ++y;
 
        // Вывод 2D массива на экран 
        printf("Rank 0's 2D array:\n");
        for (i = 0; i < ncolumns; i++)
        {
            for (j = 0; j < columnlen; j++)
                printf("%3ld ", A[i][j]);
            printf("\n");
        }
        printf("\n");
        printf("\n");
    }
    // Инициализация и выделение памяти для 1D массива столбцов на каждом процессе 
    columndata = (int*)malloc((ncolumns / p) * columnlen * sizeof(int));
    rows = (int**)malloc((ncolumns / p) * sizeof(int*));
    for (i = 0; i < (ncolumns / p); i++){
        rows[i] = &(columndata[i * columnlen]);
    }
 
    // Производный тип данных для столбцов 2D массива 
    MPI_Type_vector(columnlen, 1, 1, MPI_INT, &vector_mpi_t);
    MPI_Type_commit(&vector_mpi_t);
 
    sendbufptr = NULL;
    if (my_rank == 0) sendbufptr = &(A[0][0]);
    MPI_Scatter(sendbufptr, (ncolumns / p), vector_mpi_t, &(rows[0][0]), (ncolumns / p), vector_mpi_t, 0, MPI_COMM_WORLD);
 
    // Вывод столбцов на каждом процессе 
    printf("Rank %d's columns: \n", my_rank);
    for (i = 0; i < ncolumns / p; i++){
        for (j = 0; j < columnlen; j++){
            if ((rows[i][j] % 2 == 0 || rows[i][j] % 3 == 0 || rows[i][j] % 4 == 0 || rows[i][j] % 5 == 0 || rows[i][j] % 6 == 0 || rows[i][j] % 7 == 0
                || rows[i][j] % 8 == 0 || rows[i][j] % 9 == 0 || rows[i][j] % 10 == 0) && (rows[i][j] != 2 && rows[i][j] != 3 && rows[i][j] != 5 && rows[i][j] != 7)) {
                rows[i][j] = 0;
            }
            printf("%3ld ", rows[i][j]);
        }printf("\n");
    }
 
    // Сбор данных на 0 процессе 
    int* recvbufptr = NULL;
    if (my_rank == 0) recvbufptr = &(A[0][0]);
    MPI_Gather(&(rows[0][0]), (ncolumns / p), vector_mpi_t, recvbufptr, (ncolumns / p), vector_mpi_t, 0, MPI_COMM_WORLD);
 
    // Вывод собранных данных на 0 процессе 
    if (my_rank == 0)
    {
        printf("\nRank 0's gathered array:\n");
        for (i = 0; i < ncolumns; i++)
        {
            for (j = 0; j < columnlen; j++)
                printf("%3ld ", A[i][j]);
            printf("\n");
        }
    }
 
    MPI_Type_free(&vector_mpi_t);
    MPI_Finalize();
 
    free(rows);
    free(Adata);
    free(A);
    
    return 0;
}