ashao

program mpi_array_exchange
    use mpi
    use iso_fortran_env, only : real64

    implicit none

    integer, parameter :: num_count = 9
    integer, parameter :: num_iter = 50
    integer, dimension(num_count), parameter :: counts =[1, 2, 4, 8, 16, 32, 64, 128, 256] ! How many arrays to send
    integer, parameter :: num_elements = 64000 ! For double precision, corresponds to 512KB array

ashao

import torch
import torch.nn as nn
import numpy as np

from sklearn.model_selection import train_test_split
from sklearn import preprocessing
import torch.optim as optim 

NUM_EPOCHS = 1000
DISPLAY_EPOCH_INTERVAL = 100

ashao

import cartopy.crs as ccrs
import cartopy.feature as cfeature
import xarray as xr
import matplotlib.pyplot as plt
import matplotlib
import cmocean
import os
import json
import numpy as np
import pathlib

ashao

ashao

ashao

ashao

ashao

ashao

MODULE checksums
   !!======================================================================
   !!                       ***  MODULE checksums  ***
   !! Utility functions: Calculate checksums of arrays. Robust across PE
   !!                    count and domain decomposition
   !!
   !!======================================================================
   !! History :  3.4  ! Initial implementation  2018-04  (A.Shao)
   !!----------------------------------------------------------------------
   !!   chksum   :  Calculates a checksum of a given 2d/3d array

ashao