github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-07 22:23:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

diskcache_put

Browse Source
This commit is contained in:
Christopher Milan
2025-09-26 16:03:37 -07:00
parent c195a19c05
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: