diff --git a/fftma_module/gen/example.py b/fftma_module/gen/example.py deleted file mode 100644 index 28a3cb7..0000000 --- a/fftma_module/gen/example.py +++ /dev/null @@ -1,31 +0,0 @@ -from FFTMA import gen -import numpy as np -import sys - -N=int(sys.argv[1]) - -nx, ny, nz = N,N,N -dx, dy, dz = 1.0, 1.0, 1.0 -seed= 1548762 #rdi(10000,99999) -var=1 -vario=2 -alpha=1 -lcx=2 -lcy=4 -lcz=16 -ap1x=1 -ap1y=0 -ap1z=0 -ap2x=0 -ap2y=1 -ap2z=0 - -v1 = (var, vario, alpha, lcx, lcy, lcz, ap1x, ap1y, ap1z, ap2x, ap2y, ap2z) -variograms = [v1] - -mean=15.3245987 -variance=3.5682389 -typ=3 - -k=gen(nx, ny, nz, dx, dy, dz, seed, variograms, mean, variance, typ) -np.save(f"out_{N}.npy",k) diff --git a/fftma_module/gen/mem_profile.py b/fftma_module/gen/mem_profile.py new file mode 100644 index 0000000..55d5cba --- /dev/null +++ b/fftma_module/gen/mem_profile.py @@ -0,0 +1,16 @@ +from sympy import N +from example import generate + +from memory_profiler import memory_usage + +from memory_profiler import memory_usage +import sys + +if __name__ == '__main__': + N=int(sys.argv[1]) + + def run_gen(): + generate(N) + + mem_usage = memory_usage(run_gen) + print('Maximum memory usage: %s' % max(mem_usage)) \ No newline at end of file