Function compare_6d

test/speed_numpy.py:94–129

(n, distrib)

    print "py:hd_sd %.3f" % best_boost

def compare_6d(n, distrib):
    if distrib == 0:
        r = np.random.rand(n)
    else:
        r = 0.3 * np.random.randn(n)
    r = r.reshape(6, n/6)

    best_numpy = float("infinity")
    best_boost = float("infinity")
    for k in xrange(20):
        t = timer()
        w, xe = np.histogramdd(r.T, bins=(10, 10, 10,
                                        10, 10, 10),
                               range=((0.0, 1.0),
                                      (0.0, 1.0),
                                      (0.0, 1.0),
                                      (0.0, 1.0),
                                      (0.0, 1.0),
                                      (0.0, 1.0)))
        t = timer() - t
        best_numpy = min(t, best_numpy)

        h = histogram(regular(10, 0, 1),
                      regular(10, 0, 1),
                      regular(10, 0, 1),
                      regular(10, 0, 1),
                      regular(10, 0, 1),
                      regular(10, 0, 1))
        t = timer()
        h.fill(r[0], r[1], r[2], r[3], r[4], r[5])
        t = timer() - t
        best_boost = min(t, best_boost)
    assert(np.all(w == np.array(h)[:-2,:-2,:-2,:-2,:-2,:-2]))

    print "py:numpy %.3f" % best_numpy
    print "py:hd_sd %.3f" % best_boost

nfill = 6000000