Skip to content

Commit 0ab23a3

Browse files
committed
fixup! - Update for API changes
1 parent 894f3f9 commit 0ab23a3

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

src/model.rs

+12-4
Original file line numberDiff line numberDiff line change
@@ -612,13 +612,21 @@ impl Model {
612612
graph.add_constant(name, const_data)
613613
}
614614
Some(sg::ConstantDataType::Int8) => {
615-
let const_data =
616-
constant_data_from_storage_offset::<i8>(storage, &shape, data_offset)?;
615+
let const_data = constant_data_from_storage_offset::<i8>(
616+
storage,
617+
&shape,
618+
strides.as_deref(),
619+
data_offset,
620+
)?;
617621
graph.add_constant(name, const_data)
618622
}
619623
Some(sg::ConstantDataType::UInt8) => {
620-
let const_data =
621-
constant_data_from_storage_offset::<u8>(storage, &shape, data_offset)?;
624+
let const_data = constant_data_from_storage_offset::<u8>(
625+
storage,
626+
&shape,
627+
strides.as_deref(),
628+
data_offset,
629+
)?;
622630
graph.add_constant(name, const_data)
623631
}
624632
_ => {

0 commit comments

Comments
 (0)