From df630f8422142de13c4ba4b66e7ec4369a46f968 Mon Sep 17 00:00:00 2001
From: Praneet Rathi <prrathi10@gmail.com>
Date: Fri, 17 Jan 2025 18:26:34 -0600
Subject: [PATCH] cuda nowoprk?
---
hercules_cg/src/gpu.rs | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/hercules_cg/src/gpu.rs b/hercules_cg/src/gpu.rs
index 3e0ae81d..e7195223 100644
--- a/hercules_cg/src/gpu.rs
+++ b/hercules_cg/src/gpu.rs
@@ -1360,11 +1360,12 @@ extern \"C\" {} {}(", if ret_primitive { ret_type.clone() } else { "void".to_str
let collect_with_indices = self.codegen_collect(*collect, indices, is_char, extra_dim_collects.contains(&self.typing[collect.idx()]));
let data_variable = self.get_value(*data, false, false);
let data_type_id = self.typing[data.idx()];
- let nested_fork = nesting_fork.unwrap();
let cg_tile = match state {
KernelState::OutBlock => "grid".to_string(),
KernelState::InBlock => "block".to_string(),
- KernelState::InThread => self.get_cg_tile(nested_fork, CGType::UsePerId),
+ KernelState::InThread => {
+ self.get_cg_tile(nesting_fork.unwrap(), CGType::UsePerId)
+ }
};
if self.types[data_type_id.idx()].is_primitive() {
write!(w, "{}if ({}.thread_rank() == 0) {{\n", tabs, cg_tile)?;
--
GitLab