comp-imaging / ProxImaL

A domain-specific language for image optimization.
MIT License
112 stars 29 forks source link

Proximal codegen: Copy input images to GPU only once; copy output image back to host only once #91

Closed antonysigma closed 11 months ago

antonysigma commented 11 months ago

In the pure Halide/C++ codegen module, when the Halide-accelerated lin_op and prox_fn are computed in GPUs, mark the input data as "dirty" pending data transfer from host. Explicitly copy the output data back to host at the end of the (L-)ADMM iterations. Verify it with the total-variation denoising example.