From 9ce419ce31f0c1a7e0443d239f4a57ef2aa4ea7f Mon Sep 17 00:00:00 2001
From: Aymeric Varasse <aymeric.varasse@cea.fr>
Date: Wed, 5 Jun 2024 18:17:01 +0200
Subject: [PATCH] [nir] Fix node shape computing
---
lib/nir/node.ml | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/lib/nir/node.ml b/lib/nir/node.ml
index 94bb663..d8bb48d 100644
--- a/lib/nir/node.ml
+++ b/lib/nir/node.ml
@@ -183,7 +183,7 @@ and compute_shape_descr = function
d1 := !d1 * Shape.get shape i
done;
for i = axis to Shape.rank shape - 1 do
- d2 := !d1 * Shape.get shape i
+ d2 := !d2 * Shape.get shape i
done;
Shape.of_list [ !d1; !d2 ]
| Input { shape } -> shape
--
GitLab