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

touchup abstraction docs (#7327)

Browse source 
fix typing and use tinygrad tqdm
This commit is contained in:
chenyu 2024-10-27 22:29:55 -04:00 • committed by GitHub
commit 96fcc47e27
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
docs/abstractions2.py
View file

@ -15,8 +15,8 @@ a = MallocAllocator.alloc(4)
b = MallocAllocator.alloc(4)
# load in some values (little endian)
MallocAllocator.copyin(a, bytearray([2,0,0,0]))
MallocAllocator.copyin(b, bytearray([3,0,0,0]))
MallocAllocator.copyin(a, memoryview(bytearray([2,0,0,0])))
MallocAllocator.copyin(b, memoryview(bytearray([3,0,0,0])))
# compile a program to a binary
lib = ClangCompiler().compile("void add(int *out, int *a, int *b) { out[0] = a[0] + b[0]; }")
@ -103,7 +103,7 @@ print(sched[-1].ast)
run_schedule(sched)
# check the data out
assert out.realized.as_buffer().cast('I')[0] == 5
assert out.realized is not None and out.realized.as_buffer().cast('I')[0] == 5
print("******** fourth, the Tensor ***********")

3
docs/abstractions3.py
View file

@ -1,7 +1,6 @@
# abstractions2 goes from back to front, here we will go from front to back
from typing import List
from tqdm import tqdm
from tinygrad.helpers import DEBUG
from tinygrad.helpers import tqdm
# *****
# 0. Load mnist on the device