Merge branch 'fix_einsum' into 'main'

Fix crash in einsum

See merge request !141

.gitlab-ci.yml

-test-job:
+test-cpu:
   stage: test
   script:
     - cargo test --features=opencv
+
+test-gpu:
+  stage: test
+  script:
     - cargo test --features=cuda,opencv

hercules_ir/src/einsum.rs

@@ -235,12 +235,11 @@ impl<'a> EinsumContext<'a> {
                 control: _,
                 init: _,
                 reduct: _,
-            } => {
-                let reduce = self.so_far[&id];
+            } if let Some(reduce) = self.so_far.get(&id) => {
                 // Substitute opaque uses of thread ID nodes in inner expression
                 // with thread ID math expression, and increment inner-fork
                 // dimensions (alpha renaming).
-                return self.substitute_new_dims(reduce);
+                return self.substitute_new_dims(*reduce);
             }
             _ => MathExpr::OpaqueNode(id),
         };