Halide
12.0.1
Halide compiler and libraries
|
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.
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.