Halide 22.0.0 Halide compiler and libraries

Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
Halide::PyTorch::UserContext Struct Reference

`#include <HalidePyTorchCudaHelpers.h>`

## Public Member Functions
| UserContext (int id, CUcontext *ctx, cudaStream_t *stream)

## Public Attributes
int | device_id
CUcontext * | cuda_context
cudaStream_t * | stream

## Detailed Description

Definition at line 18 of file HalidePyTorchCudaHelpers.h.

## Constructor & Destructor Documentation

## ◆ UserContext()

Halide::PyTorch::UserContext::UserContext | ( | int | id,
|  | CUcontext * | ctx,
|  | cudaStream_t * | stream
| ) |  |
| inline

Definition at line 19 of file HalidePyTorchCudaHelpers.h.

## Member Data Documentation

## ◆ device_id

int Halide::PyTorch::UserContext::device_id

Definition at line 22 of file HalidePyTorchCudaHelpers.h.

Referenced by halide_get_gpu_device().

## ◆ cuda_context

CUcontext* Halide::PyTorch::UserContext::cuda_context

Definition at line 23 of file HalidePyTorchCudaHelpers.h.

Referenced by halide_cuda_acquire_context().

## ◆ stream

cudaStream_t* Halide::PyTorch::UserContext::stream

Definition at line 24 of file HalidePyTorchCudaHelpers.h.

Referenced by halide_cuda_get_stream().

The documentation for this struct was generated from the following file:

- src/runtime/HalidePyTorchCudaHelpers.h