Halide
HalidePyTorchCudaHelpers.h File Reference

Go to the source code of this file.

Detailed Description

Override Halide's CUDA hooks so that the Halide code called from PyTorch uses the correct GPU device and stream.

Definition in file HalidePyTorchCudaHelpers.h.