gpu: [compute] add missing error handling

Signed-off-by: Elias Naur <mail@eliasnaur.com>
2021-08-21 16:48:50 +02:00
parent 86e474688b
commit 64ccb1c06d
1 changed files with 3 additions and 1 deletions
@@ -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