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

Check that the tensor is contiguous before applying the kernel.

Browse Source
This commit is contained in:
laurent
2023-06-21 21:28:59 +01:00
parent 9834151254
commit 7c46de9584
3 changed files with 19 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/cuda_backend.rs
View File

@ -85,13 +85,15 @@ impl CudaStorage {
pub(crate) fn affine_impl(
&self,
shape: &Shape,
_stride: &[usize],
stride: &[usize],
mul: f64,
add: f64,
) -> Result<Self> {
match self {
Self::F32(arg) => {
// TODO: Handle the stride.
if !shape.is_contiguous(stride) {
todo!("affine is only implemented for the contiguous case")
}
let dev = arg.device();
let module_name = "affine_f32";
if !dev.has_func(module_name, module_name) {