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

Function SetCublasLtAttr

tensorflow/stream_executor/cuda/cuda_blas.cc:3170–3182  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

3168
3169template <typename T>
3170inline 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
3184template <typename T>
3185inline port::Status SetCublasLtAttr(cublasLtMatmulAlgo_t* handle,

Callers 4

CreateCublasLtLayoutDescFunction · 0.85
SetBiasPointerMethod · 0.85

Calls 4

ValueForStrCatFunction · 0.85
ToStringFunction · 0.70
StatusEnum · 0.50
StrCatFunction · 0.50

Tested by

no test coverage detected