proc = env.Processor(0) prgm = env.Program() code = prgm.get_stream() inp = proc.alloc_remote('f', 4, 64) out = proc.alloc_remote('f', 4, 64) out.clear() for i in xrange(0, 64): inp[i] = float(i + 1) cal.set_active_code(code) cal.dcl_input(reg.v0.x, USAGE=cal.usage.pos) cal.dcl_resource(0, cal.pixtex_type.oned, cal.fmt.float, UNNORM = True) cal.dcl_output(reg.o0, USAGE=cal.usage.generic) cal.sample(0, 0, reg.o0, reg.v0.x) prgm.set_binding(reg.i0, inp) prgm.set_binding(reg.o0, out) prgm.add(code) prgm.print_code() proc.execute(prgm) print "inp", inp print "out", out
def cal_nb_generate_2d(prgm, n_bodies, dt): code = prgm.get_stream() cal.set_active_code(code) fn_bodies = float(n_bodies) #r_cx = prgm.acquire_register() #r_cy = prgm.acquire_register() r_count = prgm.acquire_register() r_lpos = prgm.acquire_register() r_rpos = prgm.acquire_register() r_force = prgm.acquire_register() r_diff = prgm.acquire_register() r_dist_vec = prgm.acquire_register() r_dist = prgm.acquire_register() r_force_tmp = prgm.acquire_register() r_force_vec = prgm.acquire_register() r_vel = prgm.acquire_register() #code.add("dcl_input_position_interp(linear_noperspective) v0.xy__") cal.dcl_input(reg.v0.xy__, USAGE=cal.usage.pos, INTERP=cal.interp.linear_noperspective) r_bodies = prgm.acquire_register((fn_bodies,) * 4) r_G = prgm.acquire_register((G,) * 4) r_dt = prgm.acquire_register((dt,) * 4) cal.dcl_output(reg.o0, USAGE=cal.usage.generic) cal.dcl_output(reg.o1, USAGE=cal.usage.generic) cal.dcl_resource(0, cal.pixtex_type.twod, cal.fmt.float, UNNORM=True) # positions cal.dcl_resource(1, cal.pixtex_type.twod, cal.fmt.float, UNNORM=True) # velocities # Loop over all other points to calculate the force cal.mov(r_count, r_count('0000')) # loop counter #cal.mov(r_cx, r_cx('0000')) # loop counter #cal.mov(r_cy, r_cy('0000')) # loop counter cal.sample(0, 0, r_lpos, reg.v0.xy) # Local position cal.mov(r_force, r_force('0000')) # total force # Compute force using input from every other point cal.whileloop() cal.breakc(cal.relop.ge, r_count.x, r_bodies) cal.mov(r_count, r_count.x0zw) cal.whileloop() cal.breakc(cal.relop.ge, r_count.y, r_bodies) #for i in xrange(0, 4): #cal.add(r_count, r_cx('x000'), r_cy('0x00')) cal.sample(0, 0, r_rpos, r_count.xy) # Remote position # d_xyz cal.sub(r_diff, r_lpos.xyz0, r_rpos.xyz0) # local pos - remote pos # dist_tmp #cal.mul(r_dist_vec, r_diff.xxxx, r_diff.xxxx) #cal.mad(r_dist_vec, r_diff.yyyy, r_diff.yyyy, r_dist_vec) #cal.mad(r_dist_vec, r_diff.zzzz, r_diff.zzzz, r_dist_vec) cal.dp3(r_dist_vec, r_diff, r_diff, IEEE = False) # distance # TODO - skip rest of force computation if distance is 0 cal.sqrt_vec(r_dist, r_dist_vec) # force G * ((m[i] * m[j]) / dist_tmp) cal.mul(r_force_tmp, r_lpos.wwww, r_rpos.wwww, IEEE = False) cal.div(r_force_tmp, r_force_tmp, r_dist_vec, ZEROOP = cal.zeroop.zero) cal.mul(r_force_tmp, r_force_tmp, r_G, IEEE = False) # f_xyz # TODO - whats going on, is this right? cal.div(r_force_vec, r_diff.xyz0, r_dist.xyz1, ZEROOP = cal.zeroop.zero) cal.mul(r_force_vec, r_force_vec.xyz0, r_force_tmp.xyz0, IEEE = False) cal.sub(r_force, r_force.xyz0, r_force_vec.xyz0) #cal.add(r_cy, r_cy, r_count('1111')) #cal.add(r_count, r_count, r_count('0100')) #cal.ifc(cal.relop.ge, r_count.y, r_bodies.y) ## TODO - can I merge these two? #cal.mov(r_count('_y__'), r_count('x0zw')) #cal.add(r_count, r_count, r_count('1000')) #cal.endif() # Increment loop counter, end loop cal.add(r_count, r_count, r_count('0100')) cal.endloop() cal.add(r_count, r_count, r_count('1000')) #cal.add(r_cx, r_cx, r_cx('1111')) cal.endloop() # Acceleration cal.div(r_force, r_force.xyz0, r_lpos.wwww, ZEROOP = cal.zeroop.zero) # Velocity cal.sample(1, 1, r_vel, reg.v0.xy) # Load velocity cal.mad(r_vel, r_force, r_dt, r_vel, IEEE = False) cal.mov(reg.o1, r_vel) # Position cal.mad(reg.o0, r_vel.xyz0, r_dt.xyz0, r_lpos.xyzw, IEEE = False) #cal.mov(reg.g[0], r_vel) return code
def cal_nb_generate(n_bodies, dt): code = env.InstructionStream() cal.set_active_code(code) fn_bodies = float(n_bodies) r_count = code.acquire_register() r_lpos = code.acquire_register() r_rpos = code.acquire_register() r_force = code.acquire_register() r_diff = code.acquire_register() r_dist_vec = code.acquire_register() r_dist = code.acquire_register() r_force_tmp = code.acquire_register() r_force_vec = code.acquire_register() r_vel = code.acquire_register() #code.add("dcl_input_position_interp(linear_noperspective) v0.x___") cal.dcl_input(reg.v0.x___, USAGE=cal.usage.pos, INTERP=cal.interp.linear_noperspective) r_bodies = code.acquire_register((fn_bodies,) * 4) r_G = code.acquire_register((G,) * 4) r_dt = code.acquire_register((dt,) * 4) cal.dcl_output(reg.o0, USAGE=cal.usage.generic) cal.dcl_output(reg.o1, USAGE=cal.usage.generic) cal.dcl_resource(0, cal.pixtex_type.oned, cal.fmt.float, UNNORM=True) # positions cal.dcl_resource(1, cal.pixtex_type.oned, cal.fmt.float, UNNORM=True) # velocities # Loop over all other points to calculate the force cal.mov(r_count, r_count('0000')) # loop counter cal.sample(0, 0, r_lpos, reg.v0.x) # Local position cal.mov(r_force, r_force('0000')) # total force # Compute force using input from every other point cal.whileloop() # Break if end of points reached cal.breakc(cal.relop.ge, r_count, r_bodies) cal.sample(0, 0, r_rpos, r_count.x) # Remote position # d_xyz cal.sub(r_diff, r_lpos.xyz0, r_rpos.xyz0) # local pos - remote pos # dist_tmp cal.mul(r_dist_vec, r_diff.xxxx, r_diff.xxxx) cal.mad(r_dist_vec, r_diff.yyyy, r_diff.yyyy, r_dist_vec) cal.mad(r_dist_vec, r_diff.zzzz, r_diff.zzzz, r_dist_vec) # distance # TODO - skip rest of force computation if distance is 0 cal.sqrt_vec(r_dist, r_dist_vec) # force G * ((m[i] * m[j]) / dist_tmp) cal.mul(r_force_tmp, r_lpos.wwww, r_rpos.wwww) cal.div(cal.zeroop.zero, r_force_tmp, r_force_tmp, r_dist_vec) cal.mul(r_force_tmp, r_force_tmp, r_G) # f_xyz cal.div(cal.zeroop.zero, r_force_vec, r_diff.xyz0, r_dist.xyz1) cal.mul(r_force_vec, r_force_vec.xyz0, r_force_tmp.xyz0) cal.sub(r_force, r_force.xyz0, r_force_vec.xyz0) # Increment loop counter, end loop cal.add(r_count, r_count, r_count('1111')) cal.endloop() # Acceleration cal.div(cal.zeroop.zero, r_force, r_force.xyz0, r_lpos.wwww) # Velocity cal.sample(1, 1, r_vel, reg.v0.x) # Load velocity cal.mad(r_vel, r_force, r_dt, r_vel) cal.mov(reg.o1, r_vel) # Position cal.mad(reg.o0, r_vel.xyz0, r_dt.xyz0, r_lpos.xyzw) return code
proc = env.Processor(0) prgm = env.Program() code = prgm.get_stream() inp = proc.alloc_remote('f', 4, 64) out = proc.alloc_remote('f', 4, 64) out.clear() for i in xrange(0, 64): inp[i] = float(i + 1) cal.set_active_code(code) cal.dcl_input(reg.v0.x, USAGE=cal.usage.pos) cal.dcl_resource(0, cal.pixtex_type.oned, cal.fmt.float, UNNORM=True) cal.dcl_output(reg.o0, USAGE=cal.usage.generic) cal.sample(0, 0, reg.o0, reg.v0.x) prgm.set_binding(reg.i0, inp) prgm.set_binding(reg.o0, out) prgm.add(code) prgm.print_code() proc.execute(prgm) print "inp", inp print "out", out