wafer-cli 0.2.38__tar.gz → 0.2.39__tar.gz

{wafer_cli-0.2.38 → wafer_cli-0.2.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.38
+Version: 0.2.39
 Summary: CLI for running GPU workloads, managing remote workspaces, and evaluating/optimizing kernels
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{wafer_cli-0.2.38 → wafer_cli-0.2.39}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wafer-cli"
-version = "0.2.38"
+version = "0.2.39"
 description = "CLI for running GPU workloads, managing remote workspaces, and evaluating/optimizing kernels"
 readme = "README.md"
 requires-python = ">=3.11"

{wafer_cli-0.2.38 → wafer_cli-0.2.39}/wafer/agent_defaults.py

@@ -1,4 +1,4 @@
-"""Shared agent defaults for kernel optimization tasks.
+"""Shared agent defaults for kernel tasks.
 
 Single source of truth for bash allowlists and enabled tools used by both:
 - CLI templates (apps/wafer-cli/wafer/templates/*.py)
@@ -195,3 +195,50 @@ VLLM_BASH_ALLOWLIST: list[str] = [
     "cd",
     "git",
 ]
+
+# Tools available to audit agents (read-only + bash for compilation/profiling)
+AUDIT_ENABLED_TOOLS: list[str] = ["read", "glob", "grep", "bash"]
+
+# Bash commands allowed for kernel audit agents.
+AUDIT_BASH_ALLOWLIST: list[str] = [
+    # Read-only
+    "ls",
+    "cat",
+    "head",
+    "tail",
+    "wc",
+    "find",
+    "grep",
+    "rg",
+    "pwd",
+    "tree",
+    "which",
+    "diff",
+    "sort",
+    # Filesystem
+    "mkdir",
+    # Compilation
+    "make",
+    "cmake",
+    "nvcc",
+    "hipcc",
+    "g++",
+    "gcc",
+    "clang",
+    "python",
+    "python3",
+    # Execution — allows running compiled binaries via ./path.
+    # Security note: the agent can already compile arbitrary code via hipcc/gcc/etc,
+    # so blocking ./ execution doesn't add meaningful protection.
+    "./",
+    # Profiling
+    "wafer evaluate",
+    "wafer nvidia ncu",
+    "wafer nvidia nsys",
+    "wafer amd rocprof-compute",
+    "wafer amd rocprof-sdk",
+    "wafer amd rocprof-systems",
+    "wafer compiler-analyze",
+    # Misc
+    "timeout",
+]

{wafer_cli-0.2.38 → wafer_cli-0.2.39}/wafer/cli.py

@@ -1862,6 +1862,12 @@ def kernelbench_evaluate(  # noqa: PLR0913, PLR0915
         help="Sync files and generate eval script but don't run. "
         "Prints the command to run manually (useful for wrapping with rocprof, etc.)",
     ),
+    pool_timeout: int = typer.Option(
+        600,
+        "--pool-timeout",
+        help="Seconds to wait for a target from the pool before failing (default: 600). "
+        "Set to 0 for immediate failure if all targets are busy.",
+    ),
     json_output: bool = typer.Option(
         False, "--json", help="Output as single JSON object (machine-readable)"
     ),
@@ -1953,14 +1959,24 @@ def kernelbench_evaluate(  # noqa: PLR0913, PLR0915
             collector.finalize()
             raise typer.Exit(1) from None
 
-        collector.emit("pool_acquire", pool=pool, count=len(usable_targets))
-        pool_lock_context = acquire_from_pool(usable_targets)
+        effective_timeout = pool_timeout if pool_timeout > 0 else None
+        collector.emit("pool_acquire", pool=pool, count=len(usable_targets), timeout=pool_timeout)
+        pool_lock_context = acquire_from_pool(usable_targets, timeout=effective_timeout)
         acquired_target = pool_lock_context.__enter__()
 
         if acquired_target is None:
             # Exit context manager before raising to avoid resource leak
             pool_lock_context.__exit__(None, None, None)
-            collector.set_error("pool", "AllTargetsBusy", pool=pool, targets=usable_targets)
+            if pool_timeout > 0:
+                collector.set_error(
+                    "pool",
+                    "AllTargetsBusy",
+                    pool=pool,
+                    targets=usable_targets,
+                    message=f"All targets busy after waiting {pool_timeout}s",
+                )
+            else:
+                collector.set_error("pool", "AllTargetsBusy", pool=pool, targets=usable_targets)
             collector.finalize()
             raise typer.Exit(1)
 
@@ -2315,6 +2331,12 @@ def gpumode_evaluate(  # noqa: PLR0913, PLR0915
         True, "--sync-artifacts/--no-sync-artifacts", help="Download artifacts"
     ),
     gpu_id: int | None = typer.Option(None, "--gpu-id", help="Override GPU ID"),
+    pool_timeout: int = typer.Option(
+        600,
+        "--pool-timeout",
+        help="Seconds to wait for a target from the pool before failing (default: 600). "
+        "Set to 0 for immediate failure if all targets are busy.",
+    ),
 ) -> None:
     """Run kernel evaluation in GPUMode format (functional).
 
@@ -2394,14 +2416,21 @@ def gpumode_evaluate(  # noqa: PLR0913, PLR0915
             typer.echo("  Run 'wafer auth status' to see which providers need setup.", err=True)
             raise typer.Exit(1) from None
 
+        effective_timeout = pool_timeout if pool_timeout > 0 else None
         typer.echo(f"Acquiring target from pool '{pool}' ({len(usable_targets)} targets)...")
-        pool_lock_context = acquire_from_pool(usable_targets)
+        pool_lock_context = acquire_from_pool(usable_targets, timeout=effective_timeout)
         acquired_target = pool_lock_context.__enter__()
 
         if acquired_target is None:
             # Exit context manager before raising to avoid resource leak
             pool_lock_context.__exit__(None, None, None)
-            typer.echo(f"Error: All targets in pool '{pool}' are busy", err=True)
+            if pool_timeout > 0:
+                typer.echo(
+                    f"Error: All targets in pool '{pool}' are busy (waited {pool_timeout}s)",
+                    err=True,
+                )
+            else:
+                typer.echo(f"Error: All targets in pool '{pool}' are busy", err=True)
             typer.echo(f"  Targets: {', '.join(usable_targets)}", err=True)
             raise typer.Exit(1)
 
@@ -7751,16 +7780,24 @@ def compare_analyze(
         "-f",
         help="Output format: text, text-layers, csv, csv-layers, json",
     ),
-    output: Path | None = typer.Option(None, "--output", "-o", help="Output file (default: stdout)"),
+    output: Path | None = typer.Option(
+        None, "--output", "-o", help="Output file (default: stdout)"
+    ),
     phase: str = typer.Option(
         "all",
         "--phase",
         help="Filter by phase: all, prefill, decode",
     ),
     layers: bool = typer.Option(False, "--layers", help="Show layer-wise performance breakdown"),
-    all: bool = typer.Option(False, "--all", help="Show all items (no truncation for layers, operations, kernels)"),
-    stack_traces: bool = typer.Option(False, "--stack-traces", help="Show Python stack traces for operations"),
-    json: bool = typer.Option(False, "--json", hidden=True, help="Ignored (for compatibility with cliExecutor)"),
+    all: bool = typer.Option(
+        False, "--all", help="Show all items (no truncation for layers, operations, kernels)"
+    ),
+    stack_traces: bool = typer.Option(
+        False, "--stack-traces", help="Show Python stack traces for operations"
+    ),
+    json: bool = typer.Option(
+        False, "--json", hidden=True, help="Ignored (for compatibility with cliExecutor)"
+    ),
 ) -> None:
     """Compare GPU traces from two platforms platforms.
 
@@ -7824,13 +7861,17 @@ def compare_fusion_cmd(
         "-f",
         help="Output format: text, csv, json",
     ),
-    output: Path | None = typer.Option(None, "--output", "-o", help="Output file (default: stdout)"),
+    output: Path | None = typer.Option(
+        None, "--output", "-o", help="Output file (default: stdout)"
+    ),
     min_group_size: int = typer.Option(
         50,
         "--min-group-size",
         help="Minimum correlation group size to analyze",
     ),
-    json: bool = typer.Option(False, "--json", hidden=True, help="Ignored (for compatibility with cliExecutor)"),
+    json: bool = typer.Option(
+        False, "--json", hidden=True, help="Ignored (for compatibility with cliExecutor)"
+    ),
 ) -> None:
     """Analyze kernel fusion differences between AMD and NVIDIA traces.
 

{wafer_cli-0.2.38 → wafer_cli-0.2.39}/wafer/wevin_cli.py

@@ -326,13 +326,28 @@ def _build_environment(
     tools_override: list[str] | None,
     corpus_path: str | None,
     no_sandbox: bool = False,
+    has_target: bool = False,
+    template_args: dict[str, str] | None = None,
 ) -> Environment:
-    """Build a CodingEnvironment from template config."""
+    """Build a CodingEnvironment from template config.
+
+    Working directory priority:
+    1. Template arg "dir" (--args dir=./my_project) — scopes agent to a directory
+    2. corpus_path (--corpus cuda) — for doc-browsing templates
+    3. Current working directory
+    """
     from wafer_core.environments.coding import CodingEnvironment
     from wafer_core.rollouts.templates import DANGEROUS_BASH_COMMANDS
     from wafer_core.sandbox import SandboxMode
 
-    working_dir = Path(corpus_path) if corpus_path else Path.cwd()
+    # Template arg "dir" takes priority over corpus_path
+    dir_arg = (template_args or {}).get("dir")
+    if dir_arg:
+        working_dir = Path(dir_arg).resolve()
+    elif corpus_path:
+        working_dir = Path(corpus_path)
+    else:
+        working_dir = Path.cwd()
     resolved_tools = list(tools_override or tpl.tools)
 
     # Add skill tool if skills are enabled
@@ -340,12 +355,18 @@ def _build_environment(
         resolved_tools.append("skill")
 
     sandbox_mode = SandboxMode.DISABLED if no_sandbox else SandboxMode.ENABLED
+
+    # Enable network when a target is configured — the agent needs to reach
+    # remote GPUs via SSH/HTTPS. Filesystem sandbox stays enforced.
+    allow_network = has_target
+
     env: Environment = CodingEnvironment(
         working_dir=working_dir,
         enabled_tools=resolved_tools,
         bash_allowlist=tpl.bash_allowlist,
         bash_denylist=DANGEROUS_BASH_COMMANDS,
         sandbox_mode=sandbox_mode,
+        allow_network=allow_network,
     )  # type: ignore[assignment]
     return env
 
@@ -589,9 +610,21 @@ def main(  # noqa: PLR0913, PLR0915
     # CLI args override template values
     resolved_single_turn = single_turn if single_turn is not None else tpl.single_turn
 
+    # Check if a default target is configured — if so, enable network access
+    # so the agent can reach remote GPUs via SSH/HTTPS.
+    has_target = False
+    try:
+        from wafer.targets import get_default_target
+
+        has_target = get_default_target() is not None
+    except Exception:
+        pass  # No target configured — network stays disabled
+
     # Build endpoint and environment
     endpoint = _build_endpoint(tpl, model, api_base, api_key, api_key_refresh)
-    environment = _build_environment(tpl, tools, corpus_path, no_sandbox)
+    environment = _build_environment(
+        tpl, tools, corpus_path, no_sandbox, has_target=has_target, template_args=template_args
+    )
 
     # Session store
     session_store = FileSessionStore()

{wafer_cli-0.2.38 → wafer_cli-0.2.39}/wafer_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.38
+Version: 0.2.39
 Summary: CLI for running GPU workloads, managing remote workspaces, and evaluating/optimizing kernels
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown