huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 19:18:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Get the cpu backend to compile.

Browse Source
This commit is contained in:
laurent
2023-06-28 14:12:38 +01:00
parent 54a6c40f27
commit 14449ff80c
5 changed files with 44 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-core/src/tensor.rs
View File

@ -432,7 +432,7 @@ impl Tensor {
let c_shape = Shape::from(&a_dims[..dim - 2]).extend(&[m, n]);
let batching: usize = a_dims[..dim - 2].iter().product();
let storage = self.storage.matmul_impl(
let storage = self.storage.matmul(
&rhs.storage,
(batching, m, n, k),
self.layout(),
@ -587,7 +587,7 @@ impl Tensor {
}
pub fn shape(&self) -> &Shape {
&self.layout().shape()
self.layout().shape()
}
pub fn dims(&self) -> &[usize] {
@ -600,7 +600,7 @@ impl Tensor {
// TODO: Rename to `stride` once the PR that introduced the layout has been merged.
pub fn stride_tmp(&self) -> &[usize] {
&self.layout.stride()
self.layout.stride()
}
pub fn rank(&self) -> usize {