Check for nullptr when unloading model

Merge pull request #29 from wphicks/bug-failed_load
triton-inference-server · Apr 30, 2021 · 5561023 · 5561023
2 parents 375e983 + f5a2b13
commit 5561023
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");