From 7e41da1ae88435969ba8c6ed99d862a45bd4d0ac Mon Sep 17 00:00:00 2001 From: chenyu Date: Sat, 24 Jan 2026 16:47:10 -0500 Subject: [PATCH] fix generate_dataset.sh (#14324) added `set -e` so wrong pathes would fail the script, then fixed the path --- .github/workflows/test.yml | 2 +- extra/optimization/generate_dataset.sh | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index e017ab0702..178333c974 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -367,7 +367,7 @@ jobs: uses: ./.github/actions/setup-tinygrad with: key: gen-dataset - deps: testing_minimal + deps: testing opencl: 'true' - name: Generate Dataset run: CL=1 extra/optimization/generate_dataset.sh diff --git a/extra/optimization/generate_dataset.sh b/extra/optimization/generate_dataset.sh index b843dac700..87b9c06539 100755 --- a/extra/optimization/generate_dataset.sh +++ b/extra/optimization/generate_dataset.sh @@ -1,12 +1,13 @@ #!/bin/bash +set -e export PAGE_SIZE=1 export PYTHONPATH=. export LOGOPS=/tmp/ops export CAPTURE_PROCESS_REPLAY=1 -rm $LOGOPS +rm "$LOGOPS" 2>/dev/null || true test/external/process_replay/reset.py -CI=1 python3 -m pytest -n=auto test/test_ops.py test/test_nn.py test/test_winograd.py test/models/test_real_world.py --durations=20 +CI=1 python3 -m pytest -n=auto test/test_ops.py test/test_nn.py test/unit/test_winograd.py test/models/test_real_world.py --durations=20 CL=1 python3 -m pytest test/test_tiny.py # extract, sort and uniq