From 076bdef468886f56a49cb32572d9f8cd284a430a Mon Sep 17 00:00:00 2001
From: Aaron Councilman <aaronjc4@illinois.edu>
Date: Wed, 12 Feb 2025 10:54:34 -0600
Subject: [PATCH] Fix size computation for arrays
---
hercules_cg/src/gpu.rs | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/hercules_cg/src/gpu.rs b/hercules_cg/src/gpu.rs
index 01ca4773..c093949f 100644
--- a/hercules_cg/src/gpu.rs
+++ b/hercules_cg/src/gpu.rs
@@ -2008,7 +2008,11 @@ extern \"C\" {} {}(",
} else {
multiply_dcs(extents)
};
- format!("{} * {}", self.get_alignment(*element_type), array_size)
+ format!(
+ "{} * {}",
+ self.get_size(*element_type, None, extra_dim_collects),
+ array_size
+ )
}
Type::Product(fields) => {
let num_fields = num_fields.unwrap_or(fields.len());
--
GitLab