@@ -25,12 +25,12 @@ namespace cudnn_frontend::detail {
2525inline error_t
2626get_workspace_size (ManagedOpaqueDescriptor& engine_config, int64_t & workspace) {
2727#if CUDNN_VERSION >= 90200
28- CHECK_CUDNN_ERROR (detail::get_attribute (engine_config->get_backend_descriptor (),
29- CUDNN_ATTR_ENGINECFG_WORKSPACE_SIZE,
30- CUDNN_TYPE_INT64,
31- 1 ,
32- nullptr ,
33- &workspace));
28+ _CUDNN_CHECK_CUDNN_ERROR (detail::get_attribute (engine_config->get_backend_descriptor (),
29+ CUDNN_ATTR_ENGINECFG_WORKSPACE_SIZE,
30+ CUDNN_TYPE_INT64,
31+ 1 ,
32+ nullptr ,
33+ &workspace));
3434 return {error_code_t ::OK, " " };
3535#else
3636 (void )engine_config;
@@ -43,12 +43,12 @@ get_workspace_size(ManagedOpaqueDescriptor& engine_config, int64_t& workspace) {
4343inline error_t
4444get_shared_memory_size (ManagedOpaqueDescriptor& engine_config, int32_t & shared_memory_size) {
4545#if CUDNN_VERSION >= 90200
46- CHECK_CUDNN_ERROR (detail::get_attribute (engine_config->get_backend_descriptor (),
47- CUDNN_ATTR_ENGINECFG_SHARED_MEMORY_USED,
48- CUDNN_TYPE_INT32,
49- 1 ,
50- nullptr ,
51- &shared_memory_size));
46+ _CUDNN_CHECK_CUDNN_ERROR (detail::get_attribute (engine_config->get_backend_descriptor (),
47+ CUDNN_ATTR_ENGINECFG_SHARED_MEMORY_USED,
48+ CUDNN_TYPE_INT32,
49+ 1 ,
50+ nullptr ,
51+ &shared_memory_size));
5252 return {error_code_t ::OK, " " };
5353#else
5454 (void )engine_config;
@@ -63,30 +63,30 @@ create_engine(backend_descriptor& engine,
6363 int64_t const engine_id,
6464 cudnnBackendDescriptor_t op_graph,
6565 std::shared_ptr<const DeviceProperties> device_properties = nullptr ) {
66- CHECK_CUDNN_ERROR (detail::set_attribute (
66+ _CUDNN_CHECK_CUDNN_ERROR (detail::set_attribute (
6767 engine.get_ptr (), CUDNN_ATTR_ENGINE_OPERATION_GRAPH, CUDNN_TYPE_BACKEND_DESCRIPTOR, 1 , &op_graph));
6868
6969 // Validate before setting
7070 int64_t count;
71- CHECK_CUDNN_ERROR (detail::get_attribute (
71+ _CUDNN_CHECK_CUDNN_ERROR (detail::get_attribute (
7272 op_graph, CUDNN_ATTR_OPERATIONGRAPH_ENGINE_GLOBAL_COUNT, CUDNN_TYPE_INT64, 1 , nullptr , &count));
7373 RETURN_CUDNN_FRONTEND_ERROR_IF (
7474 engine_id >= count || engine_id < 0 , error_code_t ::INVALID_VALUE, " Invalid engine id." );
7575
76- CHECK_CUDNN_ERROR (
76+ _CUDNN_CHECK_CUDNN_ERROR (
7777 detail::set_attribute (engine.get_ptr (), CUDNN_ATTR_ENGINE_GLOBAL_INDEX, CUDNN_TYPE_INT64, 1 , &engine_id));
7878
7979 if (device_properties != nullptr ) {
8080#if (CUDNN_VERSION >= 90800)
81- CHECK_CUDNN_ERROR (detail::set_attribute (engine.get_ptr (),
82- CUDNN_ATTR_ENGINE_DEVICEPROP,
83- CUDNN_TYPE_BACKEND_DESCRIPTOR,
84- 1 ,
85- &device_properties->get_ptr ()));
81+ _CUDNN_CHECK_CUDNN_ERROR (detail::set_attribute (engine.get_ptr (),
82+ CUDNN_ATTR_ENGINE_DEVICEPROP,
83+ CUDNN_TYPE_BACKEND_DESCRIPTOR,
84+ 1 ,
85+ &device_properties->get_ptr ()));
8686#endif
8787 }
8888
89- CHECK_CUDNN_ERROR (detail::finalize (engine.get_ptr ()));
89+ _CUDNN_CHECK_CUDNN_ERROR (detail::finalize (engine.get_ptr ()));
9090
9191 return {error_code_t ::OK, " " };
9292}
@@ -119,37 +119,37 @@ query_knobs(int64_t const engine_id, cudnnBackendDescriptor_t op_graph, std::vec
119119
120120 // This is the actual number of knobs that is supported by the engine
121121 int64_t knobs_size;
122- CHECK_CUDNN_ERROR (detail::get_attribute (engine.get_ptr (),
123- CUDNN_ATTR_ENGINE_KNOB_INFO,
124- CUDNN_TYPE_BACKEND_DESCRIPTOR,
125- CUDNN_KNOB_TYPE_COUNTS,
126- &knobs_size,
127- backend_knobs.data ()));
122+ _CUDNN_CHECK_CUDNN_ERROR (detail::get_attribute (engine.get_ptr (),
123+ CUDNN_ATTR_ENGINE_KNOB_INFO,
124+ CUDNN_TYPE_BACKEND_DESCRIPTOR,
125+ CUDNN_KNOB_TYPE_COUNTS,
126+ &knobs_size,
127+ backend_knobs.data ()));
128128
129129 for (int64_t i = 0 ; i < knobs_size; i++) {
130130 cudnnBackendKnobType_t type;
131131 int64_t elemCount;
132- CHECK_CUDNN_ERROR (detail::get_attribute (
132+ _CUDNN_CHECK_CUDNN_ERROR (detail::get_attribute (
133133 frontend_knobs[i].get_ptr (), CUDNN_ATTR_KNOB_INFO_TYPE, CUDNN_TYPE_KNOB_TYPE, 1 , &elemCount, &type));
134134
135135 int64_t maxValue;
136- CHECK_CUDNN_ERROR (detail::get_attribute (frontend_knobs[i].get_ptr (),
137- CUDNN_ATTR_KNOB_INFO_MAXIMUM_VALUE,
138- CUDNN_TYPE_INT64,
139- 1 ,
140- &elemCount,
141- &maxValue));
136+ _CUDNN_CHECK_CUDNN_ERROR (detail::get_attribute (frontend_knobs[i].get_ptr (),
137+ CUDNN_ATTR_KNOB_INFO_MAXIMUM_VALUE,
138+ CUDNN_TYPE_INT64,
139+ 1 ,
140+ &elemCount,
141+ &maxValue));
142142
143143 int64_t minValue;
144- CHECK_CUDNN_ERROR (detail::get_attribute (frontend_knobs[i].get_ptr (),
145- CUDNN_ATTR_KNOB_INFO_MINIMUM_VALUE,
146- CUDNN_TYPE_INT64,
147- 1 ,
148- &elemCount,
149- &minValue));
144+ _CUDNN_CHECK_CUDNN_ERROR (detail::get_attribute (frontend_knobs[i].get_ptr (),
145+ CUDNN_ATTR_KNOB_INFO_MINIMUM_VALUE,
146+ CUDNN_TYPE_INT64,
147+ 1 ,
148+ &elemCount,
149+ &minValue));
150150
151151 int64_t stride;
152- CHECK_CUDNN_ERROR (detail::get_attribute (
152+ _CUDNN_CHECK_CUDNN_ERROR (detail::get_attribute (
153153 frontend_knobs[i].get_ptr (), CUDNN_ATTR_KNOB_INFO_STRIDE, CUDNN_TYPE_INT64, 1 , &elemCount, &stride));
154154
155155 auto frontend_knob_type = convert_from_backend_knob_type (type);
@@ -169,13 +169,13 @@ set_knob_choices(std::unordered_map<KnobType_t, int64_t> const& user_choices,
169169 " Failed to create knob_choice's backend descriptor." );
170170
171171 cudnnBackendKnobType_t backend_type;
172- CHECK_CUDNN_ERROR (convert_to_backend_knob_type (type, backend_type));
173- CHECK_CUDNN_ERROR (detail::set_attribute (
172+ _CUDNN_CHECK_CUDNN_ERROR (convert_to_backend_knob_type (type, backend_type));
173+ _CUDNN_CHECK_CUDNN_ERROR (detail::set_attribute (
174174 knob_choice.get_ptr (), CUDNN_ATTR_KNOB_CHOICE_KNOB_TYPE, CUDNN_TYPE_KNOB_TYPE, 1 , &backend_type));
175- CHECK_CUDNN_ERROR (detail::set_attribute (
175+ _CUDNN_CHECK_CUDNN_ERROR (detail::set_attribute (
176176 knob_choice.get_ptr (), CUDNN_ATTR_KNOB_CHOICE_KNOB_VALUE, CUDNN_TYPE_INT64, 1 , &choice));
177177
178- CHECK_CUDNN_ERROR (detail::finalize (knob_choice.get_ptr ()));
178+ _CUDNN_CHECK_CUDNN_ERROR (detail::finalize (knob_choice.get_ptr ()));
179179
180180 knob_choices.push_back (std::move (knob_choice));
181181 }
@@ -187,24 +187,24 @@ inline error_t
187187create_engine_config (ManagedOpaqueDescriptor& engine_config,
188188 backend_descriptor& engine,
189189 std::vector<detail::backend_descriptor>& knob_choices) {
190- CHECK_CUDNN_ERROR (detail::set_attribute (engine_config->get_backend_descriptor (),
191- CUDNN_ATTR_ENGINECFG_ENGINE,
192- CUDNN_TYPE_BACKEND_DESCRIPTOR,
193- 1 ,
194- &(engine.get_ptr ())));
190+ _CUDNN_CHECK_CUDNN_ERROR (detail::set_attribute (engine_config->get_backend_descriptor (),
191+ CUDNN_ATTR_ENGINECFG_ENGINE,
192+ CUDNN_TYPE_BACKEND_DESCRIPTOR,
193+ 1 ,
194+ &(engine.get_ptr ())));
195195
196196 std::vector<cudnnBackendDescriptor_t> backend_knob_choices (CUDNN_KNOB_TYPE_COUNTS);
197197 for (size_t i = 0 ; i < knob_choices.size (); i++) {
198198 backend_knob_choices[i] = knob_choices[i].get_ptr ();
199199 }
200- CHECK_CUDNN_ERROR (detail::set_attribute (engine_config->get_backend_descriptor (),
201- CUDNN_ATTR_ENGINECFG_KNOB_CHOICES,
202- CUDNN_TYPE_BACKEND_DESCRIPTOR,
203- knob_choices.size (),
204- backend_knob_choices.data ()));
200+ _CUDNN_CHECK_CUDNN_ERROR (detail::set_attribute (engine_config->get_backend_descriptor (),
201+ CUDNN_ATTR_ENGINECFG_KNOB_CHOICES,
202+ CUDNN_TYPE_BACKEND_DESCRIPTOR,
203+ knob_choices.size (),
204+ backend_knob_choices.data ()));
205205
206206 // Finalizing the descriptor
207- CHECK_CUDNN_ERROR (detail::finalize (engine_config->get_backend_descriptor ()));
207+ _CUDNN_CHECK_CUDNN_ERROR (detail::finalize (engine_config->get_backend_descriptor ()));
208208
209209 return {error_code_t ::OK, " " };
210210}
0 commit comments