github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

move print_diff to test/helpers (#7071)

Browse Source
This commit is contained in:
qazal
2024-10-15 22:00:39 +03:00
committed by GitHub
parent 1a45e94f5d
commit 09de958855
3 changed files with 18 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
test/external/process_replay/helpers.py vendored
View File

@@ -1,17 +1,7 @@
from dataclasses import dataclass
import difflib, logging, traceback, subprocess
import traceback, subprocess
from typing import Dict, Optional
from tinygrad.helpers import ContextVar, colored, getenv
def print_diff(s0, s1, unified=getenv("UNIFIED_DIFF",1)):
if not logging.getLogger().hasHandlers(): logging.basicConfig(level=logging.INFO, format="%(message)s")
if unified:
lines = list(difflib.unified_diff(str(s0).splitlines(), str(s1).splitlines()))
diff = "\n".join(colored(line, "red" if line.startswith("-") else "green" if line.startswith("+") else None) for line in lines)
else:
import ocdiff
diff = ocdiff.console_diff(str(s0), str(s1))
logging.info(diff)
from tinygrad.helpers import ContextVar, getenv
@dataclass(frozen=True)
class ProcessReplayContext:

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

@@ -5,9 +5,10 @@ from typing import Callable, List, Tuple, Union, cast
from tinygrad.engine.schedule import full_ast_rewrite
from tinygrad.helpers import VERSION, Context, ContextVar, colored, db_connection, getenv, tqdm
from tinygrad.codegen.kernel import Kernel, Opt
from test.external.process_replay.helpers import ProcessReplayContext, print_diff
from tinygrad.ops import UOp
from tinygrad.renderer import Renderer
from test.helpers import print_diff
from test.external.process_replay.helpers import ProcessReplayContext
# *** process replay settings