Selaa lähdekoodia

ggml-backend: Close on nil should be a no-op

Jesse Gross 2 kuukautta sitten
vanhempi
commit
d223f3b697
1 muutettua tiedostoa jossa 4 lisäystä ja 2 poistoa
  1. 4 2
      ml/backend/ggml/ggml.go

+ 4 - 2
ml/backend/ggml/ggml.go

@@ -327,8 +327,10 @@ func (c Context) FromIntSlice(s []int32, shape ...int) (ml.Tensor, error) {
 }
 
 func (c *Context) Close() {
-	C.ggml_backend_sched_free(c.sched)
-	C.ggml_free(c.ctx)
+	if c != nil {
+		C.ggml_backend_sched_free(c.sched)
+		C.ggml_free(c.ctx)
+	}
 }
 
 type Tensor struct {