mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-24 02:14:17 +00:00
bring buffer back to device (#4517)
This commit is contained in:
parent
a2b707a3eb
commit
d438d5698d
26 changed files with 157 additions and 174 deletions
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue