| 281 | } |
| 282 | |
| 283 | bool BlasLtMatmulEnabled() { |
| 284 | static bool is_enabled = [] { |
| 285 | bool is_enabled = false; |
| 286 | TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar( |
| 287 | "TF_USE_CUBLASLT", /*default_val=*/false, &is_enabled)); |
| 288 | return is_enabled; |
| 289 | }(); |
| 290 | return is_enabled; |
| 291 | } |
| 292 | |
| 293 | bool IsGpuCompatibleDataFormat(const RemapperContext& ctx, |
| 294 | const NodeDef* conv2d) { |
no test coverage detected