huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 10:26:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix the rnn tests for accelerate. (#704)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-01 14:21:38 +02:00
committed by GitHub
parent 7529531056
commit af552a5274
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-nn/src/rnn.rs
View File

@ -138,7 +138,8 @@ impl RNN for LSTM {
type State = LSTMState;
fn zero_state(&self, batch_dim: usize) -> Result<Self::State> {
let zeros = Tensor::zeros((batch_dim, self.hidden_dim), self.dtype, &self.device)?;
let zeros =
Tensor::zeros((batch_dim, self.hidden_dim), self.dtype, &self.device)?.contiguous()?;
Ok(Self::State {
h: zeros.clone(),
c: zeros.clone(),
@ -287,7 +288,8 @@ impl RNN for GRU {
type State = GRUState;
fn zero_state(&self, batch_dim: usize) -> Result<Self::State> {
let h = Tensor::zeros((batch_dim, self.hidden_dim), self.dtype, &self.device)?;
let h =
Tensor::zeros((batch_dim, self.hidden_dim), self.dtype, &self.device)?.contiguous()?;
Ok(Self::State { h })
}