From 405cdda7fc350900ec10985bf67a0b3ca1de6988 Mon Sep 17 00:00:00 2001
From: Russel Arbore <russel.jma@gmail.com>
Date: Sat, 1 Feb 2025 17:43:17 -0600
Subject: [PATCH 1/2] fix crash in einsum
---
hercules_ir/src/einsum.rs | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/hercules_ir/src/einsum.rs b/hercules_ir/src/einsum.rs
index c2521c22..d1c276e5 100644
--- a/hercules_ir/src/einsum.rs
+++ b/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),
};
--
GitLab
From 0abd27fee73338ed3bc772a35728d73fb4924b5e Mon Sep 17 00:00:00 2001
From: Russel Arbore <russel.jma@gmail.com>
Date: Sat, 1 Feb 2025 17:51:57 -0600
Subject: [PATCH 2/2] two separate test jobs for parallel execution
---
.gitlab-ci.yml | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 4a3ca13d..4a2a883a 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -1,5 +1,9 @@
-test-job:
+test-cpu:
stage: test
script:
- cargo test --features=opencv
+
+test-gpu:
+ stage: test
+ script:
- cargo test --features=cuda,opencv
--
GitLab