Halide  12.0.1
Halide compiler and libraries
HalidePyTorchCudaHelpers.h File Reference

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

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.