标签： gpu jit halide
我想在多个GPU平台上运行Halide Jited代码。我必须确定要使用的设备。使用用户定义设备的一般方法是替换运行时api halide_cuda_acquire_context / halide_cuda_release_context，这适用于AOT编译模式。但这不适用于Halide中的Jited代码。如何用Jited代码替换这些api？