Update app.py

app.py

@@ -1,201 +1,18 @@
-#!/usr/bin/env python3
-"""
-System Info Script — CPU Cores / Threads / Sockets + GPU Detection
-Linux ke liye (lscpu, lspci, nvidia-smi use karta hai)
-"""
+# test_chunksize.py
+import time
+from multiprocessing import Pool
 
-import subprocess
-import os
-import shutil
-
-def run_cmd(cmd):
-    """Command chalao aur output return karo (agar command na mile to None)."""
-    try:
-        result = subprocess.run(
-            cmd, shell=True, capture_output=True, text=True, timeout=10
-        )
-        if result.returncode == 0:
-            return result.stdout.strip()
-        return None
-    except Exception:
-        return None
-
-def print_section(title):
-    print("\n" + "=" * 60)
-    print(f"  {title}")
-    print("=" * 60)
-
-def check_cpu():
-    print_section("CPU Information (lscpu)")
-
-    # 1. Basic lscpu info
-    out = run_cmd("lscpu")
-    if out:
-        # Extract key lines
-        keys = [
-            "Architecture", "CPU(s)", "Thread(s) per core", "Core(s) per socket",
-            "Socket(s)", "NUMA node(s)", "Model name", "CPU MHz",
-            "CPU max MHz", "CPU min MHz", "L1d cache", "L2 cache", "L3 cache"
-        ]
-        for line in out.split("\n"):
-            for key in keys:
-                if line.strip().startswith(key):
-                    print(line.strip())
-                    break
-    else:
-        print("lscpu nahi mila. /proc/cpuinfo se try kar rahe hain...")
-        out = run_cmd("cat /proc/cpuinfo")
-        if out:
-            for line in out.split("\n"):
-                if "model name" in line or "cpu cores" in line or "siblings" in line:
-                    print(line.strip())
-
-    # 2. Physical vs Logical cores summary
-    print("\n--- Summary ---")
-    logical = os.cpu_count()
-    print(f"Logical CPUs (os.cpu_count): {logical}")
-
-    # Try to get physical cores from lscpu
-    lscpu_out = run_cmd("lscpu")
-    if lscpu_out:
-        physical = None
-        threads_per_core = None
-        sockets = None
-        for line in lscpu_out.split("\n"):
-            if "Core(s) per socket" in line:
-                try:
-                    physical = int(line.split(":")[-1].strip())
-                except: pass
-            if "Thread(s) per core" in line:
-                try:
-                    threads_per_core = int(line.split(":")[-1].strip())
-                except: pass
-            if line.strip().startswith("Socket(s)"):
-                try:
-                    sockets = int(line.split(":")[-1].strip())
-                except: pass
-
-        if physical and sockets:
-            total_physical = physical * sockets
-            print(f"Physical Cores (estimated): {total_physical}")
-            if threads_per_core:
-                print(f"Threads per Core: {threads_per_core}")
-            print(f"Hyperthreading: {'ON' if threads_per_core and threads_per_core > 1 else 'OFF'}")
-            print(f"\n💡 Recommendation: NUM_WORKERS = {total_physical} (physical cores)")
-            print(f"   Ya test karein: {total_physical // 2}, {total_physical}, {total_physical * 2}")
-
-def check_gpu():
-    print_section("GPU Information")
-
-    # 1. lspci se GPU detect karo
-    print("--- PCI Devices (lspci) ---")
-    out = run_cmd("lspci | grep -iE 'vga|3d|display'")
-    if out:
-        print(out)
-    else:
-        print("lspci se GPU detect nahi hua (ya lspci nahi hai).")
-
-    # 2. NVIDIA GPU (nvidia-smi)
-    print("\n--- NVIDIA GPU (nvidia-smi) ---")
-    if shutil.which("nvidia-smi"):
-        out = run_cmd(
-            "nvidia-smi --query-gpu=index,name,driver_version,memory.total "
-            "--format=csv,noheader"
-        )
-        if out:
-            print("Index | Name | Driver | VRAM")
-            print("-" * 50)
-            for line in out.split("\n"):
-                print(line)
-        else:
-            print("nvidia-smi present hai lekin output nahi aaya (driver issue?).")
-    else:
-        print("nvidia-smi nahi mila. NVIDIA GPU nahi hai ya driver install nahi hai.")
-
-    # 3. AMD GPU
-    print("\n--- AMD GPU (rocm-smi) ---")
-    if shutil.which("rocm-smi"):
-        out = run_cmd("rocm-smi --showproductname")
-        if out:
-            print(out)
-        else:
-            print("rocm-smi present lekin output nahi aaya.")
-    else:
-        print("rocm-smi nahi mila. AMD GPU nahi hai ya ROCm install nahi hai.")
-
-    # 4. PyTorch CUDA check
-    print("\n--- PyTorch CUDA Check ---")
-    try:
-        import torch
-        if torch.cuda.is_available():
-            print(f"✅ CUDA available: {torch.cuda.device_count()} GPU(s)")
-            for i in range(torch.cuda.device_count()):
-                print(f"   GPU {i}: {torch.cuda.get_device_name(i)}")
-                props = torch.cuda.get_device_properties(i)
-                print(f"      VRAM: {props.total_memory / 1024**3:.2f} GB")
-                print(f"      Compute Capability: {props.major}.{props.minor}")
-        else:
-            print("❌ CUDA available nahi hai (PyTorch installed hai lekin GPU nahi mila).")
-    except ImportError:
-        print("PyTorch install nahi hai (torch import fail).")
-
-def check_python_env():
-    print_section("Python Environment")
-
-    try:
-        import mnemonic
-        print(f"mnemonic: {mnemonic.__version__ if hasattr(mnemonic, '__version__') else 'installed'}")
-    except ImportError:
-        print("mnemonic: ❌ NOT installed")
-
-    try:
-        import bip39
-        print(f"bip39 (Rust): ✅ installed")
-    except ImportError:
-        print("bip39 (Rust): ❌ NOT installed — 'pip install bip39' karein (5-7x fast)")
-
-    try:
-        import coincurve
-        print(f"coincurve: ✅ installed")
-    except ImportError:
-        print("coincurve: ❌ NOT installed")
-
-    try:
-        import eth_account
-        print(f"eth_account: ✅ installed")
-    except ImportError:
-        print("eth_account: ❌ NOT installed")
-
-    import platform
-    print(f"\nPython: {platform.python_version()}")
-    print(f"OS: {platform.system()} {platform.release()}")
-
-def main():
-    print("\n" + "█" * 60)
-    print("  SYSTEM INFORMATION FOR WORKER DECISION")
-    print("█" * 60)
-
-    check_cpu()
-    check_gpu()
-    check_python_env()
-
-    print("\n" + "=" * 60)
-    print("  FINAL RECOMMENDATION")
-    print("=" * 60)
-    print("""
-1. CPU Workers: Physical cores ke barabar rakhein.
-   - 96 workers pe test karein, phir 48 aur 192 se compare karein.
-   - Hyperthreading ON hai to physical cores = logical / 2.
-
-2. GPU: Agar NVIDIA GPU hai (nvidia-smi se pata chala) to
-   CPU se 50-100x fast ho sakta hai — Hashcat ya custom CUDA use karein.
-
-3. BIP39 Speed: 'pip install bip39' (Rust) zaroor karein —
-   Python mnemonic se 5-7x fast hai.
-
-4. Aapka match ho chuka hai — ab sirf optimize karna hai to
-   ye info kaafi hai.
-""")
+def process(batch):
+    # Simple CPU kaam — 5000 phrases simulate
+    total = 0
+    for i in range(5000):
+        total += i * i
+    return total
 
 if __name__ == "__main__":
-    main()
+    for cs in [1, 2, 4, 8, 16]:
+        with Pool(48) as p:
+            start = time.time()
+            list(p.imap_unordered(process, range(2000), chunksize=cs))
+            elapsed = time.time() - start
+            print(f"chunksize={cs:2d}: {elapsed:.2f}s")