Skip to content

Commit

Permalink
clip
Browse files Browse the repository at this point in the history
  • Loading branch information
kali committed Jan 31, 2024
1 parent f1600fa commit 0d295c4
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions core/src/ops/matmul/pack.rs
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ impl TypedOp for MatMatMulPack {
mapping: &HashMap<OutletId, OutletId>,
values: &SymbolValues,
) -> TractResult<TVec<OutletId>> {
let output_shape_fact = self.output_shape_fact.eval(&values)?.into_owned();
let output_shape_fact = self.output_shape_fact.eval(values)?.into_owned();
let inputs: TVec<OutletId> = node.inputs.iter().map(|o| mapping[o]).collect();
target.wire_node(&node.name, MatMatMulPack { output_shape_fact, ..self.clone() }, &inputs)
}
Expand All @@ -100,7 +100,7 @@ impl MatMatMulPack {
let dt = input.datum_type();
unsafe {
let mut packed =
Tensor::uninitialized_aligned_dt(dt, &output_shape, self.packer.alignment())
Tensor::uninitialized_aligned_dt(dt, output_shape, self.packer.alignment())
.unwrap();
let mut bc_shape: TVec<usize> = input.shape().into();
bc_shape[self.k_axis] = 1;
Expand All @@ -118,7 +118,7 @@ impl MatMatMulPack {
prefix.remove(self.k_axis.min(self.mn_axis));
self.packer.pack(
&mut packed.view_at_prefix_mut(&prefix)?,
TensorView::from_bytes(&input, offset, input.shape(), input.strides()),
TensorView::from_bytes(input, offset, input.shape(), input.strides()),
self.k_axis,
self.mn_axis,
)
Expand Down

0 comments on commit 0d295c4

Please sign in to comment.