| 3168 | |
| 3169 | template <typename T> |
| 3170 | inline port::Status SetCublasLtAttr(cublasLtMatrixLayout_t handle, |
| 3171 | cublasLtMatrixLayoutAttribute_t attr, |
| 3172 | const T& value) { |
| 3173 | cublasStatus_t status = |
| 3174 | cublasLtMatrixLayoutSetAttribute(handle, attr, &value, sizeof(T)); |
| 3175 | if (status != CUBLAS_STATUS_SUCCESS) { |
| 3176 | return port::Status( |
| 3177 | port::error::INTERNAL, |
| 3178 | absl::StrCat("cublasLtMatrixLayoutSetAttribute(attr=", attr, |
| 3179 | ", value=", value, ") failed: ", ToString(status))); |
| 3180 | } |
| 3181 | return port::Status::OK(); |
| 3182 | } |
| 3183 | |
| 3184 | template <typename T> |
| 3185 | inline port::Status SetCublasLtAttr(cublasLtMatmulAlgo_t* handle, |
no test coverage detected