diff --git a/test/external/process_replay/process_replay.py b/test/external/process_replay/process_replay.py index 9e49369b9d..079cb4897b 100755 --- a/test/external/process_replay/process_replay.py +++ b/test/external/process_replay/process_replay.py @@ -5,7 +5,7 @@ from tinygrad.codegen.kernel import Kernel from tinygrad.helpers import Context, ContextVar, colored, db_connection, VERSION, getenv, tqdm page_size = 100 -table_name = f"process_replay_{getenv('GITHUB_SHA', 'HEAD')}_{VERSION}" +table_name = f"process_replay_{getenv('GITHUB_RUN_ID', 'HEAD')}_{VERSION}" def process_replay(offset:int): conn = db_connection() diff --git a/test/external/process_replay/reset.py b/test/external/process_replay/reset.py index 8e4f2f7011..81feae071e 100755 --- a/test/external/process_replay/reset.py +++ b/test/external/process_replay/reset.py @@ -1,4 +1,4 @@ #!/usr/bin/env python3 from tinygrad.helpers import db_connection, VERSION, getenv cur = db_connection() -cur.execute(f"drop table if exists process_replay_{getenv('GITHUB_SHA', 'HEAD')}_{VERSION}") +cur.execute(f"drop table if exists process_replay_{getenv('GITHUB_RUN_ID', 'HEAD')}_{VERSION}") diff --git a/tinygrad/codegen/kernel.py b/tinygrad/codegen/kernel.py index 2710d25543..43067ab928 100644 --- a/tinygrad/codegen/kernel.py +++ b/tinygrad/codegen/kernel.py @@ -760,7 +760,7 @@ class Kernel: self.linearize() src = self.opts.render(name:=to_function_name(self.name), self.uops) if getenv("RUN_PROCESS_REPLAY"): - table_name = f"process_replay_{getenv('GITHUB_SHA', 'HEAD')}" + table_name = f"process_replay_{getenv('GITHUB_RUN_ID', 'HEAD')}" diskcache_put(table_name, id(self), (self.ast, self.opts, self.applied_opts, name, src, {k:v.value for k,v in ContextVar._cache.items()})) info = get_lazyop_info(self.ast.src[0]) # TODO: this should be removed ops, mem = flops_mem(self.uops.uops)