| 366 | uint64_t numericImpl; |
| 367 | |
| 368 | void populate(cublasLtMatmulAlgo_t const& algo) |
| 369 | { |
| 370 | cublasLtMatmulAlgo_t const* matmulAlgo = &algo; |
| 371 | PLUGIN_CUBLASASSERT(cublasLtMatmulAlgoConfigGetAttribute( |
| 372 | matmulAlgo, CUBLASLT_ALGO_CONFIG_ID, &algoId, sizeof(algoId), nullptr)); |
| 373 | PLUGIN_CUBLASASSERT(cublasLtMatmulAlgoConfigGetAttribute( |
| 374 | matmulAlgo, CUBLASLT_ALGO_CONFIG_TILE_ID, &tile, sizeof(tile), nullptr)); |
| 375 | PLUGIN_CUBLASASSERT(cublasLtMatmulAlgoConfigGetAttribute( |
| 376 | matmulAlgo, CUBLASLT_ALGO_CONFIG_SPLITK_NUM, &numSplitsK, sizeof(numSplitsK), nullptr)); |
| 377 | PLUGIN_CUBLASASSERT(cublasLtMatmulAlgoConfigGetAttribute( |
| 378 | matmulAlgo, CUBLASLT_ALGO_CONFIG_REDUCTION_SCHEME, &reductionScheme, sizeof(reductionScheme), nullptr)); |
| 379 | PLUGIN_CUBLASASSERT(cublasLtMatmulAlgoConfigGetAttribute( |
| 380 | matmulAlgo, CUBLASLT_ALGO_CONFIG_CTA_SWIZZLING, &swizzle, sizeof(swizzle), nullptr)); |
| 381 | PLUGIN_CUBLASASSERT(cublasLtMatmulAlgoConfigGetAttribute( |
| 382 | matmulAlgo, CUBLASLT_ALGO_CONFIG_CUSTOM_OPTION, &customOption, sizeof(customOption), nullptr)); |
| 383 | PLUGIN_CUBLASASSERT(cublasLtMatmulAlgoCapGetAttribute( |
| 384 | matmulAlgo, CUBLASLT_ALGO_CAP_NUMERICAL_IMPL_FLAGS, &numericImpl, sizeof(numericImpl), nullptr)); |
| 385 | } |
| 386 | }; |
| 387 | |
| 388 | template <typename T> |
no outgoing calls
no test coverage detected