| 273 | }; |
| 274 | |
| 275 | auto build_opgraph( |
| 276 | const cudnnHandle_t& handle, const cudnnBackendDescriptorType_t operation, |
| 277 | const cudnn_frontend::Tensor& x, const cudnn_frontend::Tensor& y, |
| 278 | const cudnn_frontend::Tensor& w, const cudnn_frontend::ConvDesc_v8& conv_desc) { |
| 279 | auto op = cudnn_frontend::OperationBuilder(operation) |
| 280 | .setxDesc(x) |
| 281 | .setyDesc(y) |
| 282 | .setwDesc(w) |
| 283 | .setcDesc(conv_desc) |
| 284 | .build(); |
| 285 | std::array<cudnn_frontend::Operation const*, 1> ops = {&op}; |
| 286 | auto op_graph = cudnn_frontend::OperationGraphBuilder() |
| 287 | .setHandle(handle) |
| 288 | .setOperationGraph(1, ops.data()) |
| 289 | .build(); |
| 290 | return op_graph; |
| 291 | } |
| 292 | |
| 293 | auto build_opgraph_fused( |
| 294 | const cudnnHandle_t& handle, const cudnn_frontend::Tensor& x, |
no test coverage detected