Centre Blaise Pascal » Bench4GPU

    try:

        import matplotlib.pyplot as plt

        start_time=time.time()

        if Colors == 'Red2Yellow':

            plt.imsave("%s.png" % prefix, sigma, cmap='afmhot')

        else:

            plt.imsave("%s.png" % prefix, sigma, cmap='Greys_r')

        save_time = time.time()-start_time

        print("Save image as %s.png file" % prefix)

        print("Save Time : %f" % save_time)

    except:

        from PIL import Image

        start_time=time.time()

        Max = sigma.max()

        Min = sigma.min()

        # Normalize value as 8bits Integer

        SigmaInt = (255 * (sigma - Min) / (Max - Min)).astype("uint8")

        image = Image.fromarray(SigmaInt)

        image.save("%s.jpg" % prefix)

        save_time = time.time()-start_time

        print("Save image as %s.png file" % prefix)

        print("Save Time : %f" % save_time)

    #     TrajectoriesCL = cl.Buffer(ctx, mf.WRITE_ONLY | mf.COPY_HOST_PTR, hostbuf=Trajectories) 

    #     IdLastCL = cl.Buffer(ctx, mf.WRITE_ONLY | mf.COPY_HOST_PTR, hostbuf=IdLast)

    # zImageCL = cl.Buffer(ctx, mf.WRITE_ONLY | mf.COPY_HOST_PTR, hostbuf=zImage)

    # fImageCL = cl.Buffer(ctx, mf.WRITE_ONLY | mf.COPY_HOST_PTR, hostbuf=fImage)

        TrajectoriesCL = cl.Buffer(ctx, mf.WRITE_ONLY, Trajectories.nbytes) 

        IdLastCL = cl.Buffer(ctx, mf.WRITE_ONLY, IdLast.nbytes)

    zImageCL = cl.Buffer(ctx, mf.WRITE_ONLY, zImage.nbytes)

    fImageCL = cl.Buffer(ctx, mf.WRITE_ONLY, fImage.nbytes)