wafer-cli 0.2.1__py3-none-any.whl → 0.2.3__py3-none-any.whl

wafer/global_config.py

@@ -67,9 +67,12 @@ class Preferences:
 
     mode: "implicit" (default) = quiet output, use -v for status messages
           "explicit" = verbose output, shows [wafer] status messages
+    analytics_enabled: True (default) = send anonymous usage analytics to PostHog
+                       False = disable all analytics tracking
     """
 
     mode: Literal["explicit", "implicit"] = "implicit"
+    analytics_enabled: bool = True
 
 
 @dataclass(frozen=True)
@@ -148,7 +151,9 @@ def _parse_config_file(path: Path) -> GlobalConfig:
     pref_data = data.get("preferences", {})
     mode = pref_data.get("mode", "explicit")
     assert mode in ("explicit", "implicit"), f"mode must be 'explicit' or 'implicit', got '{mode}'"
-    preferences = Preferences(mode=mode)
+    analytics_enabled = pref_data.get("analytics_enabled", True)
+    assert isinstance(analytics_enabled, bool), f"analytics_enabled must be true or false, got '{analytics_enabled}'"
+    preferences = Preferences(mode=mode, analytics_enabled=analytics_enabled)
 
     # Parse defaults
     defaults_data = data.get("defaults", {})
@@ -301,10 +306,16 @@ def save_global_config(config: GlobalConfig) -> None:
                         lines.append(f"{key} = {value}")
                 lines.append("")
 
-    # Preferences section (only if non-default)
+    # Preferences section (only if non-default values)
+    pref_lines = []
     if config.preferences.mode != "implicit":
+        pref_lines.append(f'mode = "{config.preferences.mode}"')
+    if not config.preferences.analytics_enabled:
+        pref_lines.append("analytics_enabled = false")
+
+    if pref_lines:
         lines.append("[preferences]")
-        lines.append(f'mode = "{config.preferences.mode}"')
+        lines.extend(pref_lines)
         lines.append("")
 
     # Defaults section (only if non-default values)

wafer/skills/wafer-guide/SKILL.md

@@ -0,0 +1,116 @@
+---
+name: wafer-guide
+description: GPU kernel development with the Wafer CLI. Use when working on CUDA/HIP kernels, profiling GPU code, or optimizing kernel performance.
+---
+
+# Wafer CLI
+
+GPU development primitives for optimizing CUDA and HIP kernels.
+
+## When to Use This Skill
+
+Activate this skill when:
+- Writing or optimizing CUDA/HIP kernels
+- Profiling GPU code with NCU, NSYS, or ROCprof
+- Evaluating kernel correctness and performance
+- Looking up GPU programming documentation (CUDA, CUTLASS, HIP)
+
+## Core Workflows
+
+### 1. Documentation Lookup
+
+Query indexed GPU documentation:
+
+```bash
+# Download corpus (one-time)
+wafer corpus download cuda
+wafer corpus download cutlass
+wafer corpus download hip
+
+# Query documentation
+wafer wevin -t ask-docs --corpus cuda "What is warp divergence?"
+wafer wevin -t ask-docs --corpus cutlass "What is a TiledMma?"
+```
+
+### 2. Trace Analysis
+
+Analyze NCU, NSYS, or PyTorch profiler traces:
+
+```bash
+# AI-assisted analysis
+wafer wevin -t trace-analyze --args trace=./profile.ncu-rep "Why is this kernel slow?"
+
+# Direct trace queries (PyTorch/Perfetto JSON)
+wafer nvidia perfetto query trace.json \
+  "SELECT name, dur/1e6 as ms FROM slice WHERE cat='kernel' ORDER BY dur DESC LIMIT 10"
+
+# NCU/NSYS analysis
+wafer nvidia ncu analyze profile.ncu-rep
+wafer nvidia nsys analyze profile.nsys-rep
+```
+
+### 3. Kernel Evaluation
+
+Test correctness and measure speedup against a reference:
+
+```bash
+# Generate template files
+wafer evaluate make-template ./my-kernel
+# Creates: kernel.py, reference.py, test_cases.json
+
+# Run evaluation on a configured target
+wafer evaluate \
+  --impl ./my-kernel/kernel.py \
+  --reference ./my-kernel/reference.py \
+  --test-cases ./my-kernel/test_cases.json \
+  --target <target-name>
+
+# With profiling
+wafer evaluate ... --profile
+```
+
+### 4. AI-Assisted Optimization
+
+Iteratively optimize a kernel with evaluation feedback:
+
+```bash
+wafer wevin -t optimize-kernel \
+  --args kernel=./my_kernel.cu \
+  --args target=H100 \
+  "Optimize this GEMM for memory bandwidth"
+```
+
+### 5. Remote Execution
+
+Run on cloud GPU workspaces:
+
+```bash
+wafer workspaces list
+wafer workspaces create my-workspace --gpu H100
+wafer workspaces exec <id> "python train.py"
+wafer workspaces ssh <id>
+```
+
+## Command Reference
+
+| Command | Description |
+|---------|-------------|
+| `wafer corpus list\|download\|path` | Manage documentation corpora |
+| `wafer evaluate` | Test kernel correctness/performance |
+| `wafer nvidia ncu\|nsys\|perfetto` | NVIDIA profiling tools |
+| `wafer amd isa\|rocprof-compute` | AMD profiling tools |
+| `wafer workspaces` | Cloud GPU environments |
+| `wafer wevin -t <template>` | AI-assisted workflows |
+| `wafer config targets` | Configure GPU targets |
+
+## Target Configuration
+
+Targets define GPU access methods. Initialize with:
+
+```bash
+wafer config targets init ssh          # Your own GPU via SSH
+wafer config targets init runpod       # RunPod cloud GPUs
+wafer config targets init digitalocean # DigitalOcean AMD GPUs
+```
+
+Then use: `wafer evaluate --target <name> ...`

wafer/wevin_cli.py

@@ -420,6 +420,7 @@ def main(  # noqa: PLR0913, PLR0915
                     theme_name="minimal",
                     debug=False,
                     debug_layout=False,
+                    initial_prompt=prompt,
                 )
             else:
                 if json_output:

wafer/workspaces.py

@@ -49,6 +49,13 @@ def _friendly_error(status_code: int, response_text: str, workspace_id: str) ->
     if status_code == 401:
         return "Not authenticated. Run: wafer login"
 
+    if status_code == 402:
+        return (
+            "Insufficient credits.\n"
+            "  Check usage: wafer billing\n"
+            "  Add credits: wafer billing topup"
+        )
+
     if status_code == 404:
         return (
             f"Workspace '{workspace_id}' not found.\n"

{wafer_cli-0.2.1.dist-info → wafer_cli-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.1
+Version: 0.2.3
 Summary: CLI tool for running commands on remote GPUs and GPU kernel optimization agent
 Requires-Python: >=3.11
 Requires-Dist: typer>=0.12.0
@@ -8,6 +8,7 @@ Requires-Dist: trio>=0.24.0
 Requires-Dist: trio-asyncio>=0.15.0
 Requires-Dist: wafer-core>=0.1.0
 Requires-Dist: perfetto>=0.16.0
+Requires-Dist: posthog>=3.0.0
 Provides-Extra: dev
 Requires-Dist: pytest>=8.0.0; extra == "dev"
 Requires-Dist: pytest-cov>=4.1.0; extra == "dev"

{wafer_cli-0.2.1.dist-info → wafer_cli-0.2.3.dist-info}/RECORD

@@ -1,13 +1,15 @@
-wafer/GUIDE.md,sha256=ROnWADPya8p3YhAc7Vy9XURerIu9rZgbutNnCbpApvg,7829
+wafer/GUIDE.md,sha256=Z_jsSgHAS6bFa83VKhG9jxjUK1XpLjR1fEIKapDa_6g,3195
 wafer/__init__.py,sha256=kBM_ONCpU6UUMBOH8Tmg4A88sNFnbaD59o61cJs-uYM,90
+wafer/analytics.py,sha256=Xxw3bbY3XLgedSJPwzIOBJIjyycIiornWCpjoWbTKYU,8190
 wafer/api_client.py,sha256=cPULiTxqOAYYSfDTNJgd-6Pqrt3IM4Gm9903U7yGIwY,6163
-wafer/auth.py,sha256=3ExbDx5amwX4JiW3Ee9Vg0f0ohsS-8bqeqMIFPPE4dw,10571
+wafer/auth.py,sha256=ZLsXZ73GDLD8GL7Rij1ELtuLqyJ5EU_uPBUMPVKwExA,10703
 wafer/autotuner.py,sha256=6gH0Ho7T58EFerMQcHQxshWe3DF4qU7fb5xthAh5SPM,44364
-wafer/cli.py,sha256=pUGmg-mYHvl_S2nanj3OKrpjTeARJXOObOH_tISFBIo,160124
+wafer/billing.py,sha256=jbLB2lI4_9f2KD8uEFDi_ixLlowe5hasC0TIZJyIXRg,7163
+wafer/cli.py,sha256=kscK-JBiAu0H7roMdcaZ_lT7xlgR6mUJ1nBruidUdsE,184745
 wafer/config.py,sha256=h5Eo9_yfWqWGoPNdVQikI9GoZVUeysunSYiixf1mKcw,3411
 wafer/corpus.py,sha256=yTF3UA5bOa8BII2fmcXf-3WsIsM5DX4etysv0AzVknE,8912
 wafer/evaluate.py,sha256=7E1BqVl0um7kfU2_4FkXZrNc-PQJIHe09G1-L07PXmg,114971
-wafer/global_config.py,sha256=nQOCNjLPeXdFb0WPRpKXukXAisBNjTBw2Va6bTQGNro,10757
+wafer/global_config.py,sha256=fhaR_RU3ufMksDmOohH1OLeQ0JT0SDW1hEip_zaP75k,11345
 wafer/gpu_run.py,sha256=gUbzMsMPsw3UHcn00bI-zTSHrU8c2FEpDvbcsczlDPo,9557
 wafer/inference.py,sha256=tZCO5i05FKY27ewis3CSBHFBeFbXY3xwj0DSjdoMY9s,4314
 wafer/ncu_analyze.py,sha256=rAWzKQRZEY6E_CL3gAWUaW3uZ4kvQVZskVCPDpsFJuE,24633
@@ -17,14 +19,15 @@ wafer/rocprof_sdk.py,sha256=fAYCxpfJa5BZTTkIMBOXg4KsYK4i_wNOKrJJn1ZfypM,10086
 wafer/rocprof_systems.py,sha256=4IWbMcbYk1x_8iS7P3FC_u5sgH6EXADCtR2lV9id80M,18629
 wafer/targets.py,sha256=WE5TJgFPGtEIh7VaTQHZ4wB2t4kW0c5K8-UmQ_39Ock,10254
 wafer/tracelens.py,sha256=g9ZIeFyNojZn4uTd3skPqIrRiL7aMJOz_-GOd3aiyy4,7998
-wafer/wevin_cli.py,sha256=56xb9gbmhZKsQereF5WKR020tKp34-eVEgYAmUy5Ho8,15778
-wafer/workspaces.py,sha256=5C2U9nCuALBgHCCVBqTRY3c_JCeze9QJ104K1Im0Xgc,25570
+wafer/wevin_cli.py,sha256=tuvHM_wsJ0qo4uUV8rj93vNBL7UgDIMY6f3YXCPf4lg,15821
+wafer/workspaces.py,sha256=92LG1mtkzNz-ap3XzcqY6KnQ9SUCFG8VBIOUj1Who64,25757
+wafer/skills/wafer-guide/SKILL.md,sha256=UfBeIe5GKFzOYcbPmcs8U2nrjbfr-jSMRwg0jQDBfb0,3058
 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=u6AL7Q3uttqlnBLzcoFdsiPq5lV2TV3bgqwCYYlK9gk,2357
 wafer/templates/trace_analyze.py,sha256=XE1VqzVkIUsZbXF8EzQdDYgg-AZEYAOFpr6B_vnRELc,2880
-wafer_cli-0.2.1.dist-info/METADATA,sha256=4dKR4mpHtyumqinO0oVSEZ5HWoygIDGXY4prCHBgx8Q,529
-wafer_cli-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-wafer_cli-0.2.1.dist-info/entry_points.txt,sha256=WqB7hB__WhtPY8y1cO2sZiUz7fCq6Ik-usAigpeFvWE,41
-wafer_cli-0.2.1.dist-info/top_level.txt,sha256=2MK1IVMWfpLL8BZCQ3E9aG6L6L666gSA_teYlwan4fs,6
-wafer_cli-0.2.1.dist-info/RECORD,,
+wafer_cli-0.2.3.dist-info/METADATA,sha256=MKzRmMwc1zhKUVeu_NE4h5_17EFnX1bjLE4Jcox0AAk,559
+wafer_cli-0.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+wafer_cli-0.2.3.dist-info/entry_points.txt,sha256=WqB7hB__WhtPY8y1cO2sZiUz7fCq6Ik-usAigpeFvWE,41
+wafer_cli-0.2.3.dist-info/top_level.txt,sha256=2MK1IVMWfpLL8BZCQ3E9aG6L6L666gSA_teYlwan4fs,6
+wafer_cli-0.2.3.dist-info/RECORD,,