def square_and_sum(a, s): """ Writes np.sum(a**2,axis=0) into s """ cmin, cmax = thread_partition_array(a) map_noreturn(asqs, [(a, s, cmin[i], cmax[i]) for i in xrange(len(cmax))]) return a
def square_and_sum(a,s): """ Writes np.sum(a**2,axis=0) into s """ cmin, cmax = thread_partition_array(a) map_noreturn(asqs, [(a,s,cmin[i],cmax[i]) for i in xrange(len(cmax))]) return a
def vivax(sp_sub): cmin, cmax = thread_partition_array(sp_sub) out = sp_sub_b.copy('F') ttf = two_ten_factors[np.random.randint(len(two_ten_factors))] pm.map_noreturn(vivax_postproc, [(out, sp_sub_0, sp_sub_v, p1, ttf, cmin[i], cmax[i]) for i in xrange(len(cmax))]) return out
def vivax(sp_sub_b, sp_sub_0, sp_sub_v, p1): cmin, cmax = thread_partition_array(sp_sub_b) out = sp_sub_b.copy('F') # ttf = two_ten_factors[np.random.randint(len(two_ten_factors))] ttf = 1 pm.map_noreturn(vivax_postproc, [(out, sp_sub_0, sp_sub_v, p1, ttf, cmin[i], cmax[i]) for i in xrange(len(cmax))]) # pm.map_noreturn(vivax_postproc, [(out, sp_sub_0, sp_sub_v, p1, 1, cmin[i], cmax[i]) for i in xrange(len(cmax))]) return out
def phe0(sp_sub_b, sp_sub_0, sp_sub_v, p1): cmin, cmax = thread_partition_array(sp_sub_b) out = sp_sub_b.copy('F') pm.map_noreturn(phe0_postproc, [(out, sp_sub_0, p1, cmin[i], cmax[i]) for i in xrange(len(cmax))]) return out