Ort::Custom::RocmContext Struct Reference

#include <rocm_context.h>

Inheritance diagram for Ort::Custom::RocmContext:

Public Member Functions
void	Init (const OrtKernelContext &kernel_ctx)
Public Member Functions inherited from CustomOpContext
	CustomOpContext ()=default
virtual	~CustomOpContext ()

Public Attributes
hipStream_t	hip_stream = {}
miopenHandle_t	miopen_handle = {}
rocblas_handle	rblas_handle = {}

Detailed Description

Definition at line 16 of file rocm_context.h.

Member Function Documentation

void Ort::Custom::RocmContext::Init ( const OrtKernelContext &  kernel_ctx )

inline

Definition at line 21 of file rocm_context.h.

Member Data Documentation

hipStream_t Ort::Custom::RocmContext::hip_stream = {}

Definition at line 17 of file rocm_context.h.

miopenHandle_t Ort::Custom::RocmContext::miopen_handle = {}

Definition at line 18 of file rocm_context.h.

rocblas_handle Ort::Custom::RocmContext::rblas_handle = {}

Definition at line 19 of file rocm_context.h.

---

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

• onnxruntime/core/providers/rocm/rocm_context.h