Add a specific example for cuda.

This commit is contained in:
laurent
2023-06-21 18:56:04 +01:00
parent 2bfe8f18ab
commit c654ecdb16
3 changed files with 34 additions and 11 deletions

View File

@ -250,7 +250,7 @@ impl Tensor {
let data = S::cpu_storage_as_slice(cpu_storage)?;
Ok(self.strided_index().map(|i| data[i]).collect())
}
Storage::Cuda { .. } => todo!(),
Storage::Cuda(_) => todo!(),
}
}