heat 2d np program to generate halide schedules