Skip to content

Commit 1febb7b

Browse files
authored
Ensure output of Transpose is contiguous to prevent downstream MatMul from crashing (#3088)
1 parent 701205a commit 1febb7b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

candle-onnx/src/eval.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -443,7 +443,7 @@ fn simple_eval_(
443443
None => input.t()?,
444444
Some(perm) => {
445445
let perm = perm.iter().map(|&v| v as usize).collect::<Vec<_>>();
446-
input.permute(perm)?
446+
input.permute(perm)?.contiguous()?
447447
}
448448
};
449449
values.insert(node.output[0].clone(), output);

0 commit comments

Comments
 (0)