github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-07 22:23:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

don't allow jit input to be const (#14045)

Browse Source 
* don't allow jit input to be unbuffered like const

* just const to fix multi

* fix rnnt
This commit is contained in:
chenyu
2026-01-06 18:15:22 -05:00
committed by GitHub
parent a8896f28e1
commit c714881832
6 changed files with 21 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extra/models/rnnt.py
View File

@@ -129,7 +129,7 @@ class LSTM:
return self.do_step(x_, hc_)
if hc is None:
hc = Tensor.zeros(self.layers, 2 * x.shape[1], self.hidden_size, requires_grad=False)
hc = Tensor.zeros(self.layers, 2 * x.shape[1], self.hidden_size, requires_grad=False).contiguous().realize()
output = None
for t in range(x.shape[0]):