CUDNN Frontend API
8.3.0
|
#include <cudnn_frontend_Operation.h>
Private Types | |
using | Message_t = const char * |
Private Member Functions | |
Operation_v8 && | build_reduction_op () |
Operation_v8 && | build_matmul_op () |
Operation_v8 && | build_pointwise_op () |
Operation_v8 && | build_conv_backward_data () |
Operation_v8 && | build_conv_backward_filter () |
Operation_v8 && | build_conv_forward () |
void | extract_feature_vector (cudnnBackendDescriptorType_t op_type) |
cudnnStatus_t | validate_matmul_op (Message_t &msg) |
cudnnStatus_t | validate_reduction_op (Message_t &msg) |
cudnnStatus_t | validate_pointwise_op (Message_t &msg) |
cudnnStatus_t | validate_convolution_op (Message_t &msg) |
void | copy_dims_and_strides (const int64_t *from, int64_t *to) const |
Private Attributes | |
Operation_v8 | m_operation |
bool | is_convolution_op = false |
bool | is_pointwise_op = false |
bool | is_matmul_op = false |
bool | is_reduction_op = false |
int64_t | xTensor_dimA [CUDNN_DIM_MAX+1] |
int64_t | xTensor_strA [CUDNN_DIM_MAX+1] |
int64_t | wTensor_dimA [CUDNN_DIM_MAX+1] |
int64_t | wTensor_strA [CUDNN_DIM_MAX+1] |
int64_t | yTensor_dimA [CUDNN_DIM_MAX+1] |
int64_t | yTensor_strA [CUDNN_DIM_MAX+1] |
bool | is2D = true |
int64_t | conv_padding [CUDNN_DIM_MAX] |
int64_t | conv_dilation [CUDNN_DIM_MAX] |
int64_t | conv_stride [CUDNN_DIM_MAX] |
int64_t | mode |
int64_t | xType |
int64_t | yType |
int64_t | wType |
int64_t | cType |
int64_t | tensor_dims = 0 |
OperationBuilder_v8 Class Helper class used to build Operation_v8 class
Definition at line 156 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 164 of file cudnn_frontend_Operation.h.
|
inline |
constructs the backend Operation_v8 by calling the cudnn API Throws the appropriate error message
Definition at line 1332 of file cudnn_frontend_Operation.h.
References cudnn_frontend::getLogger(), cudnn_frontend::BackendDescriptor::initialize_managed_backend_pointer(), cudnn_frontend::Operation_v8::op_mode, cudnn_frontend::set_error_and_throw_exception(), and cudnn_frontend::BackendDescriptor::status.
|
inlineprivate |
Definition at line 556 of file cudnn_frontend_Operation.h.
References cudnn_frontend::Operation_v8::alpha_d, cudnn_frontend::Operation_v8::alpha_s, cudnn_frontend::Operation_v8::alphabetaType, cudnn_frontend::Operation_v8::beta_d, cudnn_frontend::Operation_v8::beta_s, cudnn_frontend::Operation_v8::cdesc, cudnn_frontend::Operation_v8::dxdesc, cudnn_frontend::Operation_v8::dydesc, cudnn_frontend::getLogger(), cudnn_frontend::Operation_v8::operationTag, cudnn_frontend::BackendDescriptor::pointer, cudnn_frontend::set_error_and_throw_exception(), cudnn_frontend::BackendDescriptor::status, cudnn_frontend::Operation_v8::wdesc, cudnn_frontend::Operation_v8::xdesc, and cudnn_frontend::Operation_v8::ydesc.
|
inlineprivate |
Definition at line 655 of file cudnn_frontend_Operation.h.
References cudnn_frontend::Operation_v8::alpha_d, cudnn_frontend::Operation_v8::alpha_s, cudnn_frontend::Operation_v8::alphabetaType, cudnn_frontend::Operation_v8::beta_d, cudnn_frontend::Operation_v8::beta_s, cudnn_frontend::Operation_v8::cdesc, cudnn_frontend::Operation_v8::dwdesc, cudnn_frontend::Operation_v8::dydesc, cudnn_frontend::getLogger(), cudnn_frontend::Operation_v8::operationTag, cudnn_frontend::BackendDescriptor::pointer, cudnn_frontend::set_error_and_throw_exception(), cudnn_frontend::BackendDescriptor::status, cudnn_frontend::Operation_v8::wdesc, cudnn_frontend::Operation_v8::xdesc, and cudnn_frontend::Operation_v8::ydesc.
|
inlineprivate |
Definition at line 753 of file cudnn_frontend_Operation.h.
References cudnn_frontend::Operation_v8::alpha_d, cudnn_frontend::Operation_v8::alpha_s, cudnn_frontend::Operation_v8::alphabetaType, cudnn_frontend::Operation_v8::beta_d, cudnn_frontend::Operation_v8::beta_s, cudnn_frontend::Operation_v8::cdesc, cudnn_frontend::getLogger(), cudnn_frontend::Operation_v8::operationTag, cudnn_frontend::BackendDescriptor::pointer, cudnn_frontend::set_error_and_throw_exception(), cudnn_frontend::BackendDescriptor::status, cudnn_frontend::Operation_v8::wdesc, cudnn_frontend::Operation_v8::xdesc, and cudnn_frontend::Operation_v8::ydesc.
|
inlineprivate |
Definition at line 240 of file cudnn_frontend_Operation.h.
References cudnn_frontend::Operation_v8::amatdesc, cudnn_frontend::Operation_v8::bmatdesc, cudnn_frontend::Operation_v8::cmatdesc, cudnn_frontend::Operation_v8::matmuldesc, cudnn_frontend::Operation_v8::operationTag, cudnn_frontend::BackendDescriptor::pointer, cudnn_frontend::set_error_and_throw_exception(), and cudnn_frontend::BackendDescriptor::status.
|
inlineprivate |
Definition at line 300 of file cudnn_frontend_Operation.h.
References cudnn_frontend::Operation_v8::alpha2_d, cudnn_frontend::Operation_v8::alpha2_s, cudnn_frontend::Operation_v8::alpha_d, cudnn_frontend::Operation_v8::alpha_s, cudnn_frontend::Operation_v8::alphabetaType, cudnn_frontend::Operation_v8::bdesc, cudnn_frontend::Operation_v8::dxdesc, cudnn_frontend::Operation_v8::dydesc, cudnn_frontend::Operation_v8::is_pointwise_activation_bwd_op, cudnn_frontend::Operation_v8::operationTag, cudnn_frontend::BackendDescriptor::pointer, cudnn_frontend::Operation_v8::pointwise_mode, cudnn_frontend::Operation_v8::pointwise_port_count, cudnn_frontend::Operation_v8::pwdesc, cudnn_frontend::set_error_and_throw_exception(), cudnn_frontend::BackendDescriptor::status, cudnn_frontend::Operation_v8::xdesc, and cudnn_frontend::Operation_v8::ydesc.
|
inlineprivate |
Definition at line 184 of file cudnn_frontend_Operation.h.
References cudnn_frontend::Operation_v8::operationTag, cudnn_frontend::BackendDescriptor::pointer, cudnn_frontend::Operation_v8::reductiondesc, cudnn_frontend::set_error_and_throw_exception(), cudnn_frontend::BackendDescriptor::status, cudnn_frontend::Operation_v8::xdesc, and cudnn_frontend::Operation_v8::ydesc.
|
inlineprivate |
Definition at line 1047 of file cudnn_frontend_Operation.h.
|
inlineprivate |
Build the feature vector of this operation now.
Padding
Dilation
Strides
Definition at line 845 of file cudnn_frontend_Operation.h.
References cudnn_frontend::Operation_v8::feature_vector.
|
inlineprivate |
Definition at line 978 of file cudnn_frontend_Operation.h.
References cudnn_frontend::Operation_v8::cdesc, cudnn_frontend::Operation_v8::dwdesc, cudnn_frontend::Operation_v8::dxdesc, cudnn_frontend::Operation_v8::dydesc, cudnn_frontend::Operation_v8::op_mode, cudnn_frontend::Operation_v8::wdesc, cudnn_frontend::Operation_v8::xdesc, and cudnn_frontend::Operation_v8::ydesc.
|
inlineprivate |
Definition at line 904 of file cudnn_frontend_Operation.h.
References cudnn_frontend::Operation_v8::amatdesc, cudnn_frontend::Operation_v8::bmatdesc, cudnn_frontend::Operation_v8::cmatdesc, and cudnn_frontend::Operation_v8::matmuldesc.
|
inlineprivate |
Definition at line 942 of file cudnn_frontend_Operation.h.
References cudnn_frontend::Operation_v8::bdesc, cudnn_frontend::Operation_v8::dxdesc, cudnn_frontend::Operation_v8::dydesc, cudnn_frontend::Operation_v8::is_pointwise_activation_bwd_op, cudnn_frontend::Operation_v8::is_pointwise_activation_fwd_op, cudnn_frontend::Operation_v8::is_pointwise_math_op, cudnn_frontend::Operation_v8::pointwise_port_count, cudnn_frontend::Operation_v8::xdesc, and cudnn_frontend::Operation_v8::ydesc.
|
inlineprivate |
Definition at line 925 of file cudnn_frontend_Operation.h.
References cudnn_frontend::Operation_v8::reductiondesc, cudnn_frontend::Operation_v8::xdesc, and cudnn_frontend::Operation_v8::ydesc.
|
private |
Definition at line 176 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 175 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 177 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 179 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 173 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 159 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 161 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 160 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 162 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 158 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 178 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 181 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 168 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 169 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 179 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 166 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 167 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 179 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 170 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 171 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 179 of file cudnn_frontend_Operation.h.