From de9b0bbbe4082eb8fa32760eac9962fc31aa0cbc Mon Sep 17 00:00:00 2001
From: Jared Van Bortel <jared@nomic.ai>
Date: Wed, 17 Jan 2024 10:09:27 -0500
Subject: [PATCH] add sanity check and fix kompute teardown order

---
 ggml-kompute.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ggml-kompute.cpp b/ggml-kompute.cpp
index 734f973b6..4d0dd2755 100644
--- a/ggml-kompute.cpp
+++ b/ggml-kompute.cpp
@@ -301,6 +301,7 @@ ggml_vk_device ggml_vk_current_device() {
 }
 
 ggml_kompute_context *ggml_vk_init() {
+    GGML_ASSERT(s_kompute_context == nullptr);
     s_kompute_context = new ggml_kompute_context;
     return s_kompute_context;
 }
@@ -1824,8 +1825,8 @@ static const char * ggml_backend_kompute_name(ggml_backend_t backend) {
 
 static void ggml_backend_kompute_free(ggml_backend_t backend) {
     struct ggml_kompute_context * ctx = (struct ggml_kompute_context *)backend->context;
-    ggml_vk_free_device();
     ggml_vk_free(ctx);
+    ggml_vk_free_device();
     delete backend;
 }