github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-15 17:15:48 -05:00
Code Issues Packages Projects Releases Wiki Activity

import and type cleanups [pr] (#8359)

Browse Source 
Dict and DefaultDict and some imports
This commit is contained in:
chenyu
2024-12-20 21:52:02 -05:00
committed by GitHub
parent 18dca3c3d7
commit 1ce9851ba6
7 changed files with 14 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tinygrad/runtime/ops_python.py
View File

@@ -2,9 +2,8 @@
# a python uops emulator
# works to test the tensor cores, and all the uops in general
# this is the (living) definition of uops
import sys
from typing import Optional, Any, TYPE_CHECKING
import pickle, base64, itertools, time, struct
import pickle, base64, itertools, time, struct, sys
from tinygrad.dtype import DType, dtypes, ImageDType, PtrDType, truncate
from tinygrad.helpers import all_same, getenv, flatten, get_single_element
from tinygrad.device import Compiled, Compiler, Allocator