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

hotfix: ASSERT_PROCESS_REPLAY sometimes doesn't exist (#5456)

Browse Source
This commit is contained in:
qazal
2024-07-14 02:15:40 +08:00
committed by GitHub
parent de6ab56458
commit 487ceff825
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/external/process_replay/process_replay.py vendored
View File

@@ -8,6 +8,7 @@ page_size = 100
table_name = f"process_replay_{getenv('GITHUB_RUN_ID', 'HEAD')}_{VERSION}"
def process_replay(offset:int):
ASSERT_PROCESS_REPLAY = (k:="[run_process_replay]") in os.getenv("COMMIT_MESSAGE", k) or k in os.getenv("PR_TITLE", k)
conn = db_connection()
cur = conn.cursor()
cur.execute(f"SELECT val FROM '{table_name}' LIMIT ? OFFSET ?", (page_size, offset))
@@ -40,8 +41,6 @@ def process_replay(offset:int):
cur.close()
if __name__ == "__main__":
ASSERT_PROCESS_REPLAY = (k:="[run_process_replay]") in os.getenv("COMMIT_MESSAGE", k) or k in os.getenv("PR_TITLE", k)
print(f"asserting kernel diff = {ASSERT_PROCESS_REPLAY}")
conn = db_connection()
cur = conn.cursor()
row_count = cur.execute(f"select count(*) from '{table_name}'").fetchone()[0]