From 64ccb1c06db399d45ee6ef03399c75e2a382c0e5 Mon Sep 17 00:00:00 2001
From: Elias Naur <mail@eliasnaur.com>
Date: Sat, 21 Aug 2021 16:48:50 +0200
Subject: [PATCH] gpu: [compute] add missing error handling

Signed-off-by: Elias Naur <mail@eliasnaur.com>
---
 gpu/compute.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/gpu/compute.go b/gpu/compute.go
index 1569206b..8fe104ef 100644
--- a/gpu/compute.go
+++ b/gpu/compute.go
@@ -686,7 +686,9 @@ func (g *compute) compactLayers() error {
 			continue
 		}
 		outputSize := g.output.packer.sizes[0]
-		atlas.ensureSize(g.useCPU, g.ctx, outputSize)
+		if err := atlas.ensureSize(g.useCPU, g.ctx, outputSize); err != nil {
+			return err
+		}
 		for i, l := range layers[:end] {
 			if l.newPlace == l.place {
 				continue