From 27731fa4fe3553ba1a93a98115636b60a4dd00ce Mon Sep 17 00:00:00 2001
From: Xavier Routh <xrouth2@illinois.edu>
Date: Mon, 3 Feb 2025 11:15:56 -0600
Subject: [PATCH] check for out of bounds dim on chunking
---
hercules_opt/src/fork_transforms.rs | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/hercules_opt/src/fork_transforms.rs b/hercules_opt/src/fork_transforms.rs
index cbb09bbf..190dbd25 100644
--- a/hercules_opt/src/fork_transforms.rs
+++ b/hercules_opt/src/fork_transforms.rs
@@ -725,8 +725,13 @@ pub fn chunk_fork_unguarded(
) -> () {
// tid_dim_idx = tid_dim_idx * tile_size + tid_(dim_idx + 1)
+
let Node::Fork { control: old_control, factors: ref old_factors} = *editor.node(fork) else {return};
+ if dim_idx >= old_factors.len() {
+ return; // FIXME Error here?
+ }
+
let mut new_factors: Vec<_> = old_factors.to_vec();
let fork_users: Vec<_> = editor.get_users(fork).map(|f| (f, editor.node(f).clone())).collect();
--
GitLab