[puzzles] Fix the uv install path and the sanitizer harness import

Installing the GPU Puzzles with uv works again, and puzzles that use the
out-of-bounds canary now run through the uv tasks and the
compute-sanitizer tasks instead of failing at import.

MODULAR_ORIG_COMMIT_REV_ID: d0e05670772bf8805629df62c719f30991e3e9d6

Author: KenJones-Modular

.gitignore

@@ -47,5 +47,7 @@ solutions/p16/p16_optimized
 *.ncu-rep
 *.qdrep
 *.sqlite
-*.txt
+# Profiler reports redirected to text, where `clean-profiles` looks for them.
+/*.txt
+/problems/**/*.txt
 **/p*/p*_profiler

pyproject.toml

@@ -1,5 +1,5 @@
 [build-system]
-requires = ["setuptools>=84.0.0"]
+requires = ["setuptools>=61"]
 build-backend = "setuptools.build_meta"
 
 [project]
@@ -10,12 +10,12 @@ requires-python = ">=3.10,<3.13"
 dependencies = [
     # Pinned to the release the book targets. Bump both with the release-branch
     # channel edit in `pixi.toml`; they must not drift apart again.
-    "mojo==1.0.0",
-    "max==26.5.0",
+    "mojo==1.1.0",
+    "max==26.6.0",
     "poethepoet>=0.34.0",
     "scipy>=1.15.3",
     # Matches the version `pixi.toml` pins and the suite is tested against.
-    "torch==2.14.0",
+    "torch==2.7.1",
 ]
 
 [project.optional-dependencies]
@@ -52,39 +52,39 @@ pytorch-triton-rocm = [
 [tool.poe.tasks]
 tests = "bash solutions/run.sh"
 
-p01 = "mojo problems/p01/p01.mojo"
+p01 = "mojo -I . problems/p01/p01.mojo"
 
-p02 = "mojo problems/p02/p02.mojo"
+p02 = "mojo -I . problems/p02/p02.mojo"
 
-p03 = "mojo problems/p03/p03.mojo"
+p03 = "mojo -I . problems/p03/p03.mojo"
 
-p04 = "mojo problems/p04/p04.mojo"
-p04_tile_tensor = "mojo problems/p04/p04_tile_tensor.mojo"
+p04 = "mojo -I . problems/p04/p04.mojo"
+p04_tile_tensor = "mojo -I . problems/p04/p04_tile_tensor.mojo"
 tile_tensor_intro = "mojo book/src/puzzle_04/intro.mojo"
 
-p05 = "mojo problems/p05/p05.mojo"
+p05 = "mojo -I . problems/p05/p05.mojo"
 
-p06 = "mojo problems/p06/p06.mojo"
+p06 = "mojo -I . problems/p06/p06.mojo"
 
-p07 = "mojo problems/p07/p07.mojo"
+p07 = "mojo -I . problems/p07/p07.mojo"
 
-p08 = "mojo problems/p08/p08.mojo"
+p08 = "mojo -I . problems/p08/p08.mojo"
 
 p09 = "mojo problems/p09/p09.mojo"
 
 p10 = "mojo problems/p10/p10.mojo"
 
-p11 = "mojo problems/p11/p11.mojo"
+p11 = "mojo -I . problems/p11/p11.mojo"
 
-p12 = "mojo problems/p12/p12.mojo"
+p12 = "mojo -I . problems/p12/p12.mojo"
 
-p13 = "mojo problems/p13/p13.mojo"
+p13 = "mojo -I . problems/p13/p13.mojo"
 
-p14 = "mojo problems/p14/p14.mojo"
+p14 = "mojo -I . problems/p14/p14.mojo"
 
-p15 = "mojo problems/p15/p15.mojo"
+p15 = "mojo -I . problems/p15/p15.mojo"
 
-p16 = "bash solutions/gate.sh p16 mojo problems/p16/p16.mojo"
+p16 = "bash solutions/gate.sh p16 mojo -I . problems/p16/p16.mojo"
 
 p17 = [
     { cmd = "mojo package problems/p17/op -o problems/p17/op.mojoc" },
@@ -103,35 +103,35 @@ p19 = [
 
 p20 = "python problems/p20/p20.py"
 
-p21 = "python problems/p21/p21.py"
+p21 = { cmd = "python problems/p21/p21.py", env = { PYTHONPATH = "." } }
 
-p22 = "bash solutions/gate.sh p22 python problems/p22/p22.py"
+p22 = { cmd = "bash solutions/gate.sh p22 python problems/p22/p22.py", env = { PYTHONPATH = "." } }
 
-p23 = "mojo problems/p23/p23.mojo"
+p23 = "mojo -I . problems/p23/p23.mojo"
 
-p24 = "mojo problems/p24/p24.mojo"
+p24 = "mojo -I . problems/p24/p24.mojo"
 
-p25 = "mojo problems/p25/p25.mojo"
+p25 = "mojo -I . problems/p25/p25.mojo"
 
-p26 = "mojo problems/p26/p26.mojo"
+p26 = "mojo -I . problems/p26/p26.mojo"
 
-p27 = "mojo problems/p27/p27.mojo"
+p27 = "mojo -I . problems/p27/p27.mojo"
 
-p28 = "bash solutions/gate.sh p28 mojo problems/p28/p28.mojo"
+p28 = "bash solutions/gate.sh p28 mojo -I . problems/p28/p28.mojo"
 
-p29 = "bash solutions/gate.sh p29 mojo problems/p29/p29.mojo"
+p29 = "bash solutions/gate.sh p29 mojo -I . problems/p29/p29.mojo"
 
 p30 = "mojo problems/p30/p30.mojo"
 
 p31 = "mojo problems/p31/p31.mojo"
 
 p32 = "mojo problems/p32/p32.mojo"
 
-p33 = "bash solutions/gate.sh p33 mojo problems/p33/p33.mojo"
+p33 = "bash solutions/gate.sh p33 mojo -I . problems/p33/p33.mojo"
 
-p34 = "bash solutions/gate.sh p34 mojo problems/p34/p34.mojo"
+p34 = "bash solutions/gate.sh p34 mojo -I . problems/p34/p34.mojo"
 
-p35 = "mojo problems/p35/p35.mojo"
+p35 = "mojo -I . problems/p35/p35.mojo"
 
 # GPU information
 gpu-info = { cmd = "python3 scripts/gpu_specs.py --summary" }

scripts/check_canaries.py

@@ -24,7 +24,8 @@
 kernels in ``problems/pNN/op/``, so the glob below does not reach them.
 
 Each puzzle in scope must use the canary in both its problem and solution
-files, and its `pixi` task must pass ``-I .`` so the `harness` import resolves.
+files, and both its `pixi` task and its `poe` task (the `uv` path, in
+``pyproject.toml``) must pass ``-I .`` so the `harness` import resolves.
 """
 
 from __future__ import annotations
@@ -66,8 +67,28 @@ def _pixi_tasks() -> dict[str, str]:
     )
 
 
+def _poe_tasks() -> dict[str, str]:
+    # The [tool.poe.tasks] table, parsed the same way as pixi.toml rather than
+    # with tomllib, which Python 3.10 (allowed by pyproject.toml) lacks. A
+    # task's value can be a string, an inline table or a multi-line list, so
+    # keep everything up to the next task name and search all of it.
+    text = (REPO_ROOT / "pyproject.toml").read_text()
+    table = re.search(
+        r"^\[tool\.poe\.tasks\]$(.*?)(?=^\[|\Z)", text, re.MULTILINE | re.DOTALL
+    )
+    if table is None:
+        return {}
+    return dict(
+        re.findall(
+            r"^([A-Za-z0-9_-]+) = (.*?)(?=^[A-Za-z0-9_-]+ = |\Z)",
+            table.group(1),
+            re.MULTILINE | re.DOTALL,
+        )
+    )
+
+
 def main() -> None:
-    tasks = _pixi_tasks()
+    task_files = {"pixi.toml": _pixi_tasks(), "pyproject.toml": _poe_tasks()}
     problems = sorted((REPO_ROOT / "problems").glob("p*/p*.mojo"))
     failures: list[str] = []
     checked = 0
@@ -93,14 +114,15 @@ def main() -> None:
                     )
 
         task_name = problem.stem
-        task = tasks.get(task_name)
-        if task is None:
-            failures.append(f"{rel}: no `{task_name}` task in pixi.toml")
-        elif "-I ." not in task:
-            failures.append(
-                f"pixi.toml: task `{task_name}` must pass `-I .` so the"
-                " harness import resolves"
-            )
+        for manifest, tasks in task_files.items():
+            task = tasks.get(task_name)
+            if task is None:
+                failures.append(f"{rel}: no `{task_name}` task in {manifest}")
+            elif "-I ." not in task:
+                failures.append(
+                    f"{manifest}: task `{task_name}` must pass `-I .` so the"
+                    " harness import resolves"
+                )
 
     if failures:
         print(f"Checked {checked} learner-authored puzzles.\n")

solutions/sanitizer.sh

@@ -13,6 +13,10 @@
 ##===----------------------------------------------------------------------===##
 # Source shared configuration
 SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+
+# Puzzles run from their own directory, so `mojo` needs the repo root to
+# find the `harness` package.
+PUZZLES_ROOT="$(cd "${SCRIPT_DIR}/.." && pwd)"
 source "$SCRIPT_DIR/config.sh"
 
 # Detect GPU platform and compute capability at startup
@@ -176,7 +180,7 @@ run_mojo_files_with_sanitizer() {
         # puzzle written in the third style, including the race-detection one.
         if grep -qE "(argv\(\)\[1\]|test_type|flag)[[:space:]]*==[[:space:]]*\"$specific_flag\"" "$f"; then
           echo "=== Running compute-sanitizer --tool $tool on ${path_prefix}$f with flag: $specific_flag ==="
-          output=$(compute-sanitizer --tool "$tool" mojo "$f" "$specific_flag" 2>&1)
+          output=$(compute-sanitizer --tool "$tool" mojo -I "$PUZZLES_ROOT" "$f" "$specific_flag" 2>&1)
           filtered_output=$(echo "$output" | grep -E "$grep_pattern")
 
           error_count=$(extract_error_count "$output")
@@ -203,7 +207,7 @@ run_mojo_files_with_sanitizer() {
         if [ -z "$flags" ]; then
           echo "No flags detected for ${path_prefix}$f"
           echo "=== Running compute-sanitizer --tool $tool on ${path_prefix}$f ==="
-          output=$(compute-sanitizer --tool "$tool" mojo "$f" 2>&1)
+          output=$(compute-sanitizer --tool "$tool" mojo -I "$PUZZLES_ROOT" "$f" 2>&1)
           filtered_output=$(echo "$output" | grep -E "$grep_pattern")
 
           error_count=$(extract_error_count "$output")
@@ -222,7 +226,7 @@ run_mojo_files_with_sanitizer() {
           echo "Detected flags for ${path_prefix}$f: $flags"
           for flag in $flags; do
             echo "=== Running compute-sanitizer --tool $tool on ${path_prefix}$f with flag: $flag ==="
-            output=$(compute-sanitizer --tool "$tool" mojo "$f" "$flag" 2>&1)
+            output=$(compute-sanitizer --tool "$tool" mojo -I "$PUZZLES_ROOT" "$f" "$flag" 2>&1)
             filtered_output=$(echo "$output" | grep -E "$grep_pattern")
 
             error_count=$(extract_error_count "$output")