Skip to content

GitLab

Commit 86d5cd03 authored by wenjh's avatar wenjh
Browse files

Complete manage blaslt handle 


Signed-off-by: wenjh's avatarwenjh <wenjh@sugon.com>
parent 1261da47
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
transformer_engine/common/gemm/rocm_gemm.cu
View file @ 86d5cd03
...@@ -893,10 +893,12 @@ static void CreateHipBlasLtHandle(hipblasLtHandle_t* handle) { ...@@ -893,10 +893,12 @@ static void CreateHipBlasLtHandle(hipblasLtHandle_t* handle) {
} }
static void DestroyHipBlasLtHandle(hipblasLtHandle_t handle) { static void DestroyHipBlasLtHandle(hipblasLtHandle_t handle) {
if(handle != nullptr)
NVTE_CHECK_HIPBLASLT(hipblasLtDestroy(handle)); NVTE_CHECK_HIPBLASLT(hipblasLtDestroy(handle));
}
} }
using hipBlasLtHandleManager = detail::HandleManager<hipblasLtHandle_t, CreateHipBlasLtHandle>; using hipBlasLtHandleManager = detail::HandleManager<hipblasLtHandle_t, CreateHipBlasLtHandle, DestroyHipBlasLtHandle>;
transformer_engine::DType get_transformer_engine_dtype_from_hipblaslt_dtype(const hipDataType t) { transformer_engine::DType get_transformer_engine_dtype_from_hipblaslt_dtype(const hipDataType t) {
using namespace transformer_engine; using namespace transformer_engine;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment