wafer-cli 0.2.23__py3-none-any.whl → 0.2.24__py3-none-any.whl

wafer/cli.py

@@ -4975,6 +4975,55 @@ def workspaces_sync(
         raise typer.Exit(1) from None
 
 
+@workspaces_app.command("pull")
+def workspaces_pull(
+    workspace: str = typer.Argument(..., help="Workspace name or ID"),
+    remote_path: str = typer.Argument(..., help="Remote path in workspace (relative to /workspace or absolute)"),
+    local_path: Path = typer.Argument(
+        Path("."), help="Local destination path (default: current directory)"
+    ),
+    verbose: bool = typer.Option(False, "--verbose", "-v", help="Show [wafer] status messages"),
+    quiet: bool = typer.Option(False, "--quiet", "-q", help="Suppress [wafer] status messages"),
+) -> None:
+    """Pull files from workspace to local machine.
+
+    Uses rsync over SSH to download files from the workspace's /workspace directory.
+
+    Examples:
+        wafer workspaces pull dev kernel.py ./           # Pull single file
+        wafer workspaces pull dev kernel.py ./my_kernel.py  # Pull and rename
+        wafer workspaces pull dev /workspace/results ./  # Pull directory
+    """
+    from .global_config import get_preferences
+    from .workspaces import pull_files
+
+    # Determine verbosity based on mode
+    prefs = get_preferences()
+    if quiet:
+        show_status = False
+    elif verbose:
+        show_status = True
+    else:
+        show_status = prefs.mode == "explicit"
+
+    if show_status:
+        typer.echo(f"[wafer] Pulling {remote_path} from workspace {workspace}...", err=True)
+
+    def on_progress(msg: str) -> None:
+        if show_status:
+            typer.echo(f"[wafer] {msg}", err=True)
+
+    try:
+        file_count = pull_files(
+            workspace, remote_path, local_path.resolve(), on_progress=on_progress
+        )
+        if show_status:
+            typer.echo(f"[wafer] Pulled {file_count} files to {local_path}", err=True)
+    except RuntimeError as e:
+        typer.echo(f"Error: {e}", err=True)
+        raise typer.Exit(1) from None
+
+
 # =============================================================================
 # Target operations commands (exec/ssh/sync)
 # =============================================================================

wafer/evaluate.py

@@ -2801,15 +2801,6 @@ if torch.cuda.is_available():
     gc.collect()
     torch.cuda.empty_cache()
     torch.cuda.reset_peak_memory_stats()
-    
-    # Enable TF32 for fair benchmarking against reference kernels.
-    # PyTorch 1.12+ disables TF32 for matmul by default, which handicaps
-    # reference kernels using cuBLAS. We enable it so reference kernels
-    # run at their best performance (using tensor cores when applicable).
-    # This ensures speedup comparisons are against optimized baselines.
-    torch.backends.cuda.matmul.allow_tf32 = True
-    torch.backends.cudnn.allow_tf32 = True
-    print("[KernelBench] TF32 enabled for fair benchmarking")
 
 
 def _calculate_timing_stats(times: list[float]) -> dict:

wafer/wevin_cli.py

@@ -205,10 +205,22 @@ def _get_session_preview(session: object) -> str:
     return ""
 
 
+def _get_log_file_path() -> Path:
+    """Get user-specific log file path, creating directory if needed.
+    
+    Uses ~/.wafer/logs/ to avoid permission issues with shared /tmp.
+    """
+    log_dir = Path.home() / ".wafer" / "logs"
+    log_dir.mkdir(parents=True, exist_ok=True)
+    return log_dir / "wevin_debug.log"
+
+
 def _setup_logging() -> None:
     """Configure logging to file only (no console spam)."""
     import logging.config
 
+    log_file = _get_log_file_path()
+
     logging.config.dictConfig({
         "version": 1,
         "disable_existing_loggers": False,
@@ -220,7 +232,7 @@ def _setup_logging() -> None:
         "handlers": {
             "file": {
                 "class": "logging.handlers.RotatingFileHandler",
-                "filename": "/tmp/wevin_debug.log",
+                "filename": str(log_file),
                 "maxBytes": 10_000_000,
                 "backupCount": 3,
                 "formatter": "json",

wafer/workspaces.py

@@ -542,6 +542,102 @@ def sync_files(
     return file_count, warning
 
 
+def pull_files(
+    workspace_id: str,
+    remote_path: str,
+    local_path: Path,
+    on_progress: Callable[[str], None] | None = None,
+) -> int:
+    """Pull files from workspace to local via rsync over SSH.
+
+    Args:
+        workspace_id: Workspace ID or name
+        remote_path: Remote path in workspace (relative to /workspace or absolute)
+        local_path: Local destination path
+        on_progress: Optional callback for progress messages
+
+    Returns:
+        Number of files transferred
+
+    Raises:
+        RuntimeError: If rsync fails or workspace not accessible
+    """
+    import subprocess
+
+    def emit(msg: str) -> None:
+        if on_progress:
+            on_progress(msg)
+
+    assert workspace_id, "Workspace ID must be non-empty"
+
+    ws = get_workspace_raw(workspace_id)
+    workspace_status = ws.get("status")
+    assert workspace_status in VALID_STATUSES, (
+        f"Workspace {workspace_id} has invalid status '{workspace_status}'. "
+        f"Valid statuses: {VALID_STATUSES}"
+    )
+    if workspace_status == "error":
+        raise RuntimeError(
+            f"Workspace provisioning failed. Delete and recreate:\n"
+            f"  wafer workspaces delete {workspace_id}\n"
+            f"  wafer workspaces create {ws.get('name', workspace_id)} --wait"
+        )
+    if workspace_status != "running":
+        raise RuntimeError(
+            f"Workspace is {workspace_status}. Wait for it to be running before pulling files."
+        )
+    ssh_host = ws.get("ssh_host")
+    ssh_port = ws.get("ssh_port")
+    ssh_user = ws.get("ssh_user")
+    if not ssh_host or not ssh_port or not ssh_user:
+        raise RuntimeError(
+            f"Workspace is running but SSH not ready.\n"
+            f"  Delete and recreate: wafer workspaces delete {workspace_id}\n"
+            f"  Then: wafer workspaces create {ws.get('name', workspace_id)} --wait"
+        )
+    assert isinstance(ssh_port, int) and ssh_port > 0, "Workspace missing valid ssh_port"
+
+    # Normalize remote path - if not absolute, assume relative to /workspace
+    if not remote_path.startswith("/"):
+        remote_path = f"/workspace/{remote_path}"
+
+    # Build SSH command for rsync
+    ssh_opts = f"-p {ssh_port} -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null"
+
+    # Build rsync command (reverse of sync - from remote to local)
+    rsync_cmd = [
+        "rsync",
+        "-avz",
+        "-e",
+        f"ssh {ssh_opts}",
+        f"{ssh_user}@{ssh_host}:{remote_path}",
+        str(local_path),
+    ]
+
+    emit(f"Pulling {remote_path} from workspace...")
+
+    try:
+        result = subprocess.run(rsync_cmd, capture_output=True, text=True)
+        if result.returncode != 0:
+            raise RuntimeError(f"rsync failed: {result.stderr}")
+
+        # Count files from rsync output
+        lines = result.stdout.strip().split("\n")
+        file_count = sum(
+            1
+            for line in lines
+            if line and not line.startswith((" ", "sent", "total", "receiving", "building"))
+        )
+
+    except FileNotFoundError:
+        raise RuntimeError("rsync not found. Install rsync to use pull feature.") from None
+    except subprocess.SubprocessError as e:
+        raise RuntimeError(f"Pull failed: {e}") from e
+
+    emit(f"Pulled {file_count} files")
+    return file_count
+
+
 def _init_sync_state(workspace_id: str) -> str | None:
     """Tell API to sync files from bare metal to Modal volume.
 

{wafer_cli-0.2.23.dist-info → wafer_cli-0.2.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.23
+Version: 0.2.24
 Summary: CLI tool for running commands on remote GPUs and GPU kernel optimization agent
 Requires-Python: >=3.11
 Requires-Dist: typer>=0.12.0

{wafer_cli-0.2.23.dist-info → wafer_cli-0.2.24.dist-info}/RECORD

@@ -5,10 +5,10 @@ wafer/api_client.py,sha256=i_Az2b2llC3DSW8yOL-BKqa7LSKuxOr8hSN40s-oQXY,6313
 wafer/auth.py,sha256=dwss_se5P-FFc9IN38q4kh_dBrA6k-CguDBkivgcdj0,14003
 wafer/autotuner.py,sha256=41WYP41pTDvMijv2h42vm89bcHtDMJXObDlWmn6xpFU,44416
 wafer/billing.py,sha256=jbLB2lI4_9f2KD8uEFDi_ixLlowe5hasC0TIZJyIXRg,7163
-wafer/cli.py,sha256=j4ODOVT_r-kyc21YOI8Yl8bkiZMGuqDpXRs7CvpNaek,261443
+wafer/cli.py,sha256=tk7wnmwZiqlOT39E3EVVwgD-2Y0eAfLSAbif7IpLijY,263288
 wafer/config.py,sha256=h5Eo9_yfWqWGoPNdVQikI9GoZVUeysunSYiixf1mKcw,3411
 wafer/corpus.py,sha256=oQegXA43MuyRvYxOsWhmqeP5vMb5IKFHOvM-1RcahPA,22301
-wafer/evaluate.py,sha256=SxxhiPkO6aDdfktRzJXpbWMVmIGn_gw-o5C6Zwj2zRc,190930
+wafer/evaluate.py,sha256=I0JwipvkkOIiiYaIyqe-HMSfYdTPkUtJXvs_1w_gJSQ,190413
 wafer/global_config.py,sha256=fhaR_RU3ufMksDmOohH1OLeQ0JT0SDW1hEip_zaP75k,11345
 wafer/gpu_run.py,sha256=TwqXy72T7f2I7e6n5WWod3xgxCPnDhU0BgLsB4CUoQY,9716
 wafer/inference.py,sha256=tZCO5i05FKY27ewis3CSBHFBeFbXY3xwj0DSjdoMY9s,4314
@@ -26,16 +26,16 @@ wafer/target_lock.py,sha256=SDKhNzv2N7gsphGflcNni9FE5YYuAMuEthngAJEo4Gs,7809
 wafer/targets.py,sha256=9r-iRWoKSH5cQl1LcamaX-T7cNVOg99ngIm_hlRk-qU,26922
 wafer/targets_ops.py,sha256=jN1oIBx0mutxRNE9xpIc7SaBxPkVmOyus2eqn0kEKNI,21475
 wafer/tracelens.py,sha256=g9ZIeFyNojZn4uTd3skPqIrRiL7aMJOz_-GOd3aiyy4,7998
-wafer/wevin_cli.py,sha256=Nuk7zTCiJrnpmYtdg5Hu0NbzONCqs54xtON6K7AVB9U,23189
-wafer/workspaces.py,sha256=iUdioK7kA3z_gOTMNVDn9Q87c6qpkdXF4bOhJWkUPg8,32375
+wafer/wevin_cli.py,sha256=UwRibTYmIR99EjDupoyFm8gjhMs0vHXTkS4wQuiskho,23537
+wafer/workspaces.py,sha256=k_iCZ-mOrG2KiTXqqcZ5_VifSIXsFGaZM4hjnxBnBmc,35666
 wafer/skills/wafer-guide/SKILL.md,sha256=KWetJw2TVTbz11_nzqazqOJWWRlbHRFShs4sOoreiWo,3255
 wafer/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wafer/templates/ask_docs.py,sha256=Lxs-faz9v5m4Qa4NjF2X_lE8KwM9ES9MNJkxo7ep56o,2256
 wafer/templates/optimize_kernel.py,sha256=OvZgN5tm_OymO3lK8Dr0VO48e-5PfNVIIoACrPxpmqk,2446
 wafer/templates/optimize_kernelbench.py,sha256=aoOA13zWEl89r6QW03xF9NKxQ7j4mWe9rwua6-mlr4Y,4780
 wafer/templates/trace_analyze.py,sha256=XE1VqzVkIUsZbXF8EzQdDYgg-AZEYAOFpr6B_vnRELc,2880
-wafer_cli-0.2.23.dist-info/METADATA,sha256=LS8QcHjtver7DSzO75TAfJWH2H7gNswXzAp7Av4_OCU,560
-wafer_cli-0.2.23.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-wafer_cli-0.2.23.dist-info/entry_points.txt,sha256=WqB7hB__WhtPY8y1cO2sZiUz7fCq6Ik-usAigpeFvWE,41
-wafer_cli-0.2.23.dist-info/top_level.txt,sha256=2MK1IVMWfpLL8BZCQ3E9aG6L6L666gSA_teYlwan4fs,6
-wafer_cli-0.2.23.dist-info/RECORD,,
+wafer_cli-0.2.24.dist-info/METADATA,sha256=zoGLJsbN_xgLNwaGRnUll2HY545hFXoBYwYtzeGmlvo,560
+wafer_cli-0.2.24.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+wafer_cli-0.2.24.dist-info/entry_points.txt,sha256=WqB7hB__WhtPY8y1cO2sZiUz7fCq6Ik-usAigpeFvWE,41
+wafer_cli-0.2.24.dist-info/top_level.txt,sha256=2MK1IVMWfpLL8BZCQ3E9aG6L6L666gSA_teYlwan4fs,6
+wafer_cli-0.2.24.dist-info/RECORD,,