From f5a2b13ba9313d249cccc2e526af29594b72d5b6 Mon Sep 17 00:00:00 2001
From: William Hicks <whicks@nvidia.com>
Date: Wed, 28 Apr 2021 12:18:49 -0400
Subject: [PATCH] Check for nullptr when unloading model

---
 src/api.cu | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/api.cu b/src/api.cu
index 0327e268..69551242 100644
--- 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");