diff --git a/extra/datasets/sops.gz b/extra/datasets/sops.gz index 3aa9d8fe5d..60daacea06 100644 Binary files a/extra/datasets/sops.gz and b/extra/datasets/sops.gz differ diff --git a/extra/optimization/generate_dataset.sh b/extra/optimization/generate_dataset.sh index cc421bc821..747a74d5a2 100755 --- a/extra/optimization/generate_dataset.sh +++ b/extra/optimization/generate_dataset.sh @@ -12,9 +12,9 @@ WINO=1 STEPS=3 python3 examples/hlb_cifar10.py python3 examples/stable_diffusion.py --noshow python3 examples/llama.py --prompt "hello" --count 5 python3 examples/gpt2.py --count 5 -python3 HALF=1 examples/gpt2.py --count 5 -python3 python examples/beautiful_mnist.py -python3 python examples/beautiful_cartpole.py +HALF=1 python3 examples/gpt2.py --count 5 +python3 examples/beautiful_mnist.py +python3 examples/beautiful_cartpole.py python3 examples/mlperf/model_spec.py python3 examples/yolov8.py ./test/models/efficientnet/Chicken.jpg openpilot/go.sh diff --git a/tinygrad/realize.py b/tinygrad/realize.py index beb15a479e..d189941deb 100644 --- a/tinygrad/realize.py +++ b/tinygrad/realize.py @@ -50,7 +50,7 @@ logops = open(getenv("LOGOPS", ""), "a") if getenv("LOGOPS", "") else None def run_schedule(schedule:List[ScheduleItem]): while len(schedule): si = schedule.pop(0) - if logops and si.ast.op not in LoadOps: logops.write(str(si.ast)+"\n") + if logops and si.ast.op not in LoadOps and not any(i.device.startswith("DISK:") for i in si.inputs): logops.write(str(si.ast)+"\n") # get the program prg = lower_schedule_item(si)