bring buffer back to device (#4517)

This commit is contained in:
George Hotz 2024-05-10 11:22:31 -07:00 committed by GitHub
commit d438d5698d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
26 changed files with 157 additions and 174 deletions

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