Check for nullptr when unloading model

triton-inference-server · Apr 28, 2021 · f5a2b13 · f5a2b13
1 parent 4196033
commit f5a2b13
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/api.cu b/src/api.cu
@@ -97,7 +97,9 @@ TRITONBACKEND_ModelFinalize(TRITONBACKEND_Model* model)
 {
   try {
     auto model_state = get_model_state<ModelState>(*model);
-    model_state->UnloadModel();
+    if (model_state != nullptr) {
+      model_state->UnloadModel();
+    }
 
     LOG_MESSAGE(
         TRITONSERVER_LOG_INFO, "TRITONBACKEND_ModelFinalize: delete model state");