MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / BlasLtMatmulEnabled

Function BlasLtMatmulEnabled

tensorflow/core/grappler/optimizers/remapper.cc:283–291  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

281}
282
283bool 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
293bool IsGpuCompatibleDataFormat(const RemapperContext& ctx,
294 const NodeDef* conv2d) {

Callers 1

IsGpuCompatibleMatMulFunction · 0.85

Calls 1

ReadBoolFromEnvVarFunction · 0.50

Tested by

no test coverage detected