mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

remove obsolete TODO in beautiful_mnist (#2946)

Browse source 
the compiler error was due to `error: call to 'max' is ambiguous` when we have max(int, float) in kernel.
it was first fixed in 4380ccb1 the non fp32 math PR, and further solidified with dtype refactor
This commit is contained in:
chenyu 2023-12-28 17:09:23 -05:00 • committed by GitHub
commit 2f67f1e580
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
examples/beautiful_mnist.py
View file

@ -23,7 +23,6 @@ if __name__ == "__main__":
model = Model()
opt = nn.optim.Adam(nn.state.get_parameters(model))
# TODO: there's a compiler error if you comment out TinyJit since randint isn't being realized and there's something weird with int
@TinyJit
def train_step(samples:Tensor) -> Tensor:
with Tensor.train():