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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.39
+Version: 0.2.40
 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.39 → wafer_cli-0.2.40}/pyproject.toml

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

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

@@ -220,16 +220,22 @@ workspaces_app = typer.Typer(
 
 Workspaces are on-demand cloud GPU environments. Requires authentication (wafer login).
 
+Environment Types:
+  modal      Serverless GPU execution (fast startup, no SSH)
+  baremetal  Dedicated GPU server (SSH access, hardware counters)
+
 Available GPUs:
-  MI300X  AMD Instinct MI300X (192GB HBM3, ROCm)
+  MI300X  AMD Instinct MI300X (192GB HBM3, ROCm) - baremetal only
   B200    NVIDIA Blackwell B200 (180GB HBM3e, CUDA)
+  H100    NVIDIA Hopper H100 (80GB HBM3, CUDA)
 
 Commands:
-  wafer workspaces create dev --gpu B200   # Create workspace
-  wafer workspaces exec dev -- python x.py # Run commands
-  wafer workspaces ssh dev                 # Interactive SSH
-  wafer workspaces sync dev ./project      # Sync files
-  wafer workspaces delete dev              # Clean up"""
+  wafer workspaces create dev -g B200 -e modal      # Create Modal workspace
+  wafer workspaces create dev -g B200 -e baremetal  # Create baremetal workspace
+  wafer workspaces exec dev -- python x.py          # Run commands
+  wafer workspaces ssh dev                          # Interactive SSH (baremetal only)
+  wafer workspaces sync dev ./project               # Sync files
+  wafer workspaces delete dev                       # Clean up"""
 )
 app.add_typer(workspaces_app, name="workspaces")
 
@@ -3272,7 +3278,7 @@ def demo_eval(
         # Step 1: Create workspace
         typer.echo(f"\n[1/4] Creating workspace '{workspace_name}'...")
         result = subprocess.run(
-            ["wafer", "workspaces", "create", workspace_name, "--gpu", "B200", "--json"],
+            ["wafer", "workspaces", "create", workspace_name, "--gpu", "B200", "--environment", "modal", "--json"],
             capture_output=True,
             text=True,
             check=True,
@@ -3359,7 +3365,7 @@ print(f"Performance: {(t1-t0)/100*1e6:.1f} us/iter")
             typer.echo("\n✓ Demo complete! To evaluate your own kernels:")
             typer.echo("")
             typer.echo("  # Using workspaces (no setup required):")
-            typer.echo("  wafer workspaces create dev --gpu B200")
+            typer.echo("  wafer workspaces create dev --gpu B200 --environment modal")
             typer.echo("  wafer workspaces exec --sync ./my-kernel dev -- python my_test.py")
             typer.echo("")
             typer.echo("  # Or using wafer evaluate with a configured target:")
@@ -4816,7 +4822,10 @@ def workspaces_list(
 def workspaces_create(
     name: str = typer.Argument(..., help="Workspace name"),
     gpu_type: str = typer.Option(
-        "B200", "--gpu", "-g", help="GPU type: MI300X (AMD) or B200 (NVIDIA, default)"
+        ..., "--gpu", "-g", help="GPU type: MI300X (AMD) or B200/H100 (NVIDIA)"
+    ),
+    environment: str = typer.Option(
+        ..., "--environment", "-e", help="Environment type: modal or baremetal"
     ),
     image: str | None = typer.Option(None, "--image", "-i", help="Docker image (optional)"),
     wait: bool = typer.Option(
@@ -4826,16 +4835,22 @@ def workspaces_create(
 ) -> None:
     """Create a new workspace.
 
+    Per-vendor architecture: each workspace has a single environment type.
+
+    Environment Types:
+        modal      Serverless GPU execution (fast startup, no SSH)
+        baremetal  Dedicated GPU server (SSH access, hardware counters)
+
     Available GPUs:
-        MI300X  AMD Instinct MI300X (192GB HBM3, ROCm)
-        B200    NVIDIA Blackwell B200 (180GB HBM3e, CUDA)
+        MI300X     AMD Instinct MI300X (192GB HBM3, ROCm) - baremetal only
+        B200       NVIDIA Blackwell B200 (180GB HBM3e, CUDA)
+        H100       NVIDIA Hopper H100 (80GB HBM3, CUDA)
 
     Example:
-        wafer workspaces create my-kernel                # B200 (default)
-        wafer workspaces create my-kernel --gpu MI300X   # AMD MI300X
-        wafer workspaces create my-kernel --gpu B200     # NVIDIA B200
-        wafer workspaces create my-kernel --image pytorch/pytorch:2.5.1-cuda12.4-cudnn9-devel
-        wafer workspaces create my-kernel --wait
+        wafer workspaces create my-kernel --gpu B200 --environment modal
+        wafer workspaces create my-kernel --gpu MI300X --environment baremetal
+        wafer workspaces create my-kernel -g B200 -e baremetal  # SSH + ncu profiling
+        wafer workspaces create my-kernel -g B200 -e modal --wait
     """
     from .workspaces import create_workspace
 
@@ -4843,6 +4858,7 @@ def workspaces_create(
         result = create_workspace(
             name,
             gpu_type=gpu_type,
+            environment_type=environment,
             image=image,
             wait=wait,
             json_output=json_output,

{wafer_cli-0.2.39 → wafer_cli-0.2.40}/wafer/workspaces.py

@@ -249,16 +249,20 @@ def list_workspaces(json_output: bool = False) -> str:
 
 def create_workspace(
     name: str,
-    gpu_type: str = "B200",
+    gpu_type: str,
+    environment_type: str,
     image: str | None = None,
     wait: bool = False,
     json_output: bool = False,
 ) -> str:
     """Create a new workspace.
 
+    Per-vendor architecture: each workspace has a single environment type.
+
     Args:
         name: Workspace name (must be unique)
-        gpu_type: GPU type (default: B200)
+        gpu_type: GPU type (required: B200, H100, MI300X)
+        environment_type: Environment type (required: modal, baremetal)
         image: Docker image (optional, uses default if not specified)
         wait: If True, stream provisioning progress and return SSH credentials
         json_output: If True, return raw JSON; otherwise return formatted text
@@ -272,6 +276,7 @@ def create_workspace(
     # Validate inputs
     assert name, "Workspace name must be non-empty"
     assert gpu_type, "GPU type must be non-empty"
+    assert environment_type, "Environment type must be non-empty"
 
     api_url, headers = _get_client()
 
@@ -296,6 +301,7 @@ def create_workspace(
     request_body: dict = {
         "name": name,
         "gpu_type": gpu_type,
+        "environment_type": environment_type,
     }
     if image:
         request_body["image"] = image

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

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