如何使用自动调度程序为GPU生成时间表？

Question

我想在SIGGRAPH'16发表的论文中重新检查有关GPU自动调度的实验。

我使用auto-scheduler分支编译了下面的代码，但生成的代码是为CPU预定的。

如何使用自动调度程序为GPU生成时间表？

#include "Halide.h" 
#include "HalideBuffer.h" 
#include <climits> 
#include <iostream> 

using namespace Halide; 

class conv2dHalide : public Halide::Generator<conv2dHalide>{ 
public: 
    ImageParam in{UInt(8), 2, "input_image"}; 
    ImageParam kernel{Float(32), 2, "kernel"}; 

    Var x, y; 

    Func build(){ 
     Func out("output_image"); 

    // Algorithm 
     Func clamped_in; 
     Expr k_wid = 3; 
     Expr k_half = k_wid/2; 
     Expr i_wid = in.width(); 
     Expr i_hei = in.height(); 
     Expr c_x = clamp(x, 0, in.width()-1); 
     Expr c_y = clamp(y, 0, in.height()-1); 

     clamped_in(x,y) = in(c_x,c_y); 

     RDom r(-k_half, k_wid, -k_half, k_wid); 
     r.where(k_half<=x && x<i_wid-k_half && k_half<=y && y<i_hei-k_half); 
     out(x,y)=cast<uint8_t>(min(sum(clamped_in(x+r.x, y+r.y)*kernel(r.x+k_half, r.y+k_half)), 255.0f)); 

     out.estimate(x, 0, 512).estimate(y, 0, 512); 
     in.dim(0).set_bounds_estimate(0, i_wid); 
     in.dim(1).set_bounds_estimate(0, i_hei); 
     kernel.dim(0).set_bounds_estimate(0, k_wid); 
     kernel.dim(1).set_bounds_estimate(0, k_wid); 

     Target target = Halide::get_host_target(); 
     target.set_feature(Target::CUDA); 
     target.set_feature(Target::CUDACapability50); 
     std::cout << "gpu enabled" << std::endl; 

     Pipeline p(out); 
     p.auto_schedule(target); 
     out.print_loop_nest(); 

     return out; 
    } 

}; 

HALIDE_REGISTER_GENERATOR(conv2dHalide, "conv2dHalide") 

Answer 1

我们（现在）在让自动调度程序准备好合并到主线Halide的同时削减GPU的东西。我认为你必须回到他用于SIGGRAPH纸拉维的卤化物叉：

https://github.com/ravi-teja-mullapudi/Halide

我想看看在应用程序的例子，并就回购打开一个问题，如果你是无法将其安排到GPU中。