Halide
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 20 of file HalidePyTorchCudaHelpers.h.

◆ cuda_context

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

Definition at line 23 of file HalidePyTorchCudaHelpers.h.

◆ stream

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

Definition at line 24 of file HalidePyTorchCudaHelpers.h.

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