Skip to content

Commit c7bd5fb

Browse files
aborgna-qss2165
andauthored
fix: hugr-llvm extension deltas (#1683)
Fixes #1680 --------- Co-authored-by: Seyon Sivarajah <seyon.sivarajah@quantinuum.com>
1 parent 4145024 commit c7bd5fb

File tree

2 files changed

+22
-3
lines changed

2 files changed

+22
-3
lines changed

hugr-llvm/src/emit/test.rs

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,13 @@ use anyhow::{anyhow, Result};
44
use hugr_core::builder::{
55
BuildHandle, Container, DFGWrapper, HugrBuilder, ModuleBuilder, SubContainer,
66
};
7-
use hugr_core::extension::{ExtensionRegistry, EMPTY_REG};
7+
use hugr_core::extension::prelude::PRELUDE_ID;
8+
use hugr_core::extension::{ExtensionRegistry, ExtensionSet, EMPTY_REG};
89
use hugr_core::ops::handle::FuncID;
10+
use hugr_core::std_extensions::arithmetic::{
11+
conversions, float_ops, float_types, int_ops, int_types,
12+
};
13+
use hugr_core::std_extensions::logic;
914
use hugr_core::types::TypeRow;
1015
use hugr_core::{Hugr, HugrView};
1116
use inkwell::module::Module;
@@ -137,7 +142,20 @@ impl SimpleHugrConfig {
137142
) -> Hugr {
138143
let mut mod_b = ModuleBuilder::new();
139144
let func_b = mod_b
140-
.define_function("main", HugrFuncType::new(self.ins, self.outs))
145+
.define_function(
146+
"main",
147+
HugrFuncType::new(self.ins, self.outs).with_extension_delta(
148+
ExtensionSet::from_iter([
149+
PRELUDE_ID,
150+
int_types::EXTENSION_ID,
151+
int_ops::EXTENSION_ID,
152+
float_types::EXTENSION_ID,
153+
float_ops::EXTENSION_ID,
154+
conversions::EXTENSION_ID,
155+
logic::EXTENSION_ID,
156+
]),
157+
),
158+
)
141159
.unwrap();
142160
make(func_b, &self.extensions);
143161

hugr-llvm/src/utils/array_op_builder.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,7 @@ impl<D: Dataflow> ArrayOpBuilder for D {}
115115

116116
#[cfg(test)]
117117
pub mod test {
118+
use hugr_core::extension::prelude::PRELUDE_ID;
118119
use hugr_core::{
119120
builder::{DFGBuilder, HugrBuilder},
120121
extension::{
@@ -133,7 +134,7 @@ pub mod test {
133134
#[rstest::fixture]
134135
#[default(DFGBuilder<Hugr>)]
135136
pub fn all_array_ops<B: Dataflow>(
136-
#[default(DFGBuilder::new(Signature::new_endo(Type::EMPTY_TYPEROW)).unwrap())]
137+
#[default(DFGBuilder::new(Signature::new_endo(Type::EMPTY_TYPEROW).with_prelude()).unwrap())]
137138
mut builder: B,
138139
) -> B {
139140
let us0 = builder.add_load_value(ConstUsize::new(0));

0 commit comments

Comments
 (0)