from __future__ import division
import pyopencl as cl
import pyopencl.characterize.performance as perf




def main():
    ctx = cl.create_some_context()

    prof_overhead, latency = perf.get_profiling_overhead(ctx)
    print "command latency: %g s" % latency
    print "profiling overhead: %g s -> %.1f %%" % (
            prof_overhead, 100*prof_overhead/latency)
    queue = cl.CommandQueue(ctx, properties=cl.command_queue_properties.PROFILING_ENABLE)

    print "empty kernel: %g s" % perf.get_empty_kernel_time(queue)
    print "float32 add: %g GOps/s" % (perf.get_add_rate(queue)/1e9)

    for tx_type in [
            perf.HostToDeviceTransfer,
            perf.DeviceToHostTransfer,
            perf.DeviceToDeviceTransfer]:
        print "----------------------------------------"
        print tx_type.__name__
        print "----------------------------------------"

        print "latency: %g s" % perf.transfer_latency(queue, tx_type)
        for i in range(6, 28, 2):
            bs = 1<<i
            print "bandwidth @ %d bytes: %g GB/s" % (
                    bs, perf.transfer_bandwidth(queue, tx_type, bs)/1e9)




if __name__ == "__main__":
    main()