geohot/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Actions Activity

bring buffer back to device (#4517)

Browse Source
This commit is contained in:
George Hotz
2024-05-10 11:22:31 -07:00
committed by GitHub
parent a2b707a3eb
commit d438d5698d
26 changed files with 157 additions and 174 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/ops_python.py
+1 -2
View File
@@ -5,8 +5,7 @@ from typing import Tuple, List, Optional, Any, Dict
import pickle, base64, itertools, time, struct
from tinygrad.dtype import DType, dtypes, ImageDType
from tinygrad.helpers import all_same, getenv, flatten
from tinygrad.device import Compiled, Compiler, CompilerOptions
from tinygrad.buffer import Allocator
from tinygrad.device import Compiled, Compiler, CompilerOptions, Allocator
from tinygrad.codegen.uops import UOpGraph, UOps
from tinygrad.ops import BinaryOps, TernaryOps, exec_alu