Complete manage blaslt handle

Signed-off-by: wenjh <wenjh@sugon.com>

Complete manage blaslt handle
Signed-off-by: wenjh <wenjh@sugon.com>
86d5cd03 · wenjh · 1261da47 · 86d5cd03
Commit 86d5cd03 authored Dec 15, 2025 by wenjh
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

transformer_engine/common/gemm/rocm_gemm.cu transformer_engine/common/gemm/rocm_gemm.cu +4 -2

No files found.
--- a/transformer_engine/common/gemm/rocm_gemm.cu
+++ b/transformer_engine/common/gemm/rocm_gemm.cu
@@ -893,10 +893,12 @@ static void CreateHipBlasLtHandle(hipblasLtHandle_t* handle) {
 }

 static void DestroyHipBlasLtHandle(hipblasLtHandle_t handle) {
-  NVTE_CHECK_HIPBLASLT(hipblasLtDestroy(handle));
+  if(handle != nullptr)
+    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) {
  using namespace transformer_engine;