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

diskcache_put

Browse source
This commit is contained in:
Christopher Milan 2025-09-26 16:03:37 -07:00
commit 930fa3de8a
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tinygrad/helpers.py
View file

@ -262,9 +262,9 @@ def diskcache_get(table:str, key:dict|str|bytes|int) -> Any:
return None
_db_tables = set()
def diskcache_put(table:str, key:dict|str|int, val:Any, prepickled=False):
def diskcache_put(table:str, key:dict|str|bytes|int, val:Any, prepickled=False):
if CACHELEVEL < 1: return val
if isinstance(key, (str,int)): key = {"key": key}
if isinstance(key, (str,bytes,int)): key = {"key": key}
conn = db_connection()
cur = conn.cursor()
if table not in _db_tables: