@adverant-nexus/cli 2.0.0

package/dist/commands/compute/lib/hardware-detection.js

@@ -0,0 +1,432 @@
+/**
+ * Hardware Detection Module
+ *
+ * Detects local hardware capabilities including:
+ * - Apple Silicon (M1/M2/M3/M4 series)
+ * - NVIDIA GPUs
+ * - CPU cores and memory
+ * - Installed ML frameworks
+ */
+import os from 'os';
+import { execSync } from 'child_process';
+/**
+ * Apple Silicon GPU Specifications
+ */
+const APPLE_SILICON_SPECS = {
+    'Apple M1': {
+        memory: 16,
+        fp32Tflops: 2.6,
+        fp16Tflops: 5.2,
+        neuralEngine: true,
+        neuralEngineTops: 11,
+        api: 'Metal 3',
+    },
+    'Apple M1 Pro': {
+        memory: 32,
+        fp32Tflops: 5.2,
+        fp16Tflops: 10.4,
+        neuralEngine: true,
+        neuralEngineTops: 11,
+        api: 'Metal 3',
+    },
+    'Apple M1 Max': {
+        memory: 64,
+        fp32Tflops: 10.4,
+        fp16Tflops: 20.8,
+        neuralEngine: true,
+        neuralEngineTops: 11,
+        api: 'Metal 3',
+    },
+    'Apple M1 Ultra': {
+        memory: 128,
+        fp32Tflops: 20.8,
+        fp16Tflops: 41.6,
+        neuralEngine: true,
+        neuralEngineTops: 22,
+        api: 'Metal 3',
+    },
+    'Apple M2': {
+        memory: 24,
+        fp32Tflops: 3.6,
+        fp16Tflops: 7.2,
+        neuralEngine: true,
+        neuralEngineTops: 15.8,
+        api: 'Metal 3',
+    },
+    'Apple M2 Pro': {
+        memory: 32,
+        fp32Tflops: 6.8,
+        fp16Tflops: 13.6,
+        neuralEngine: true,
+        neuralEngineTops: 15.8,
+        api: 'Metal 3',
+    },
+    'Apple M2 Max': {
+        memory: 96,
+        fp32Tflops: 13.6,
+        fp16Tflops: 27.2,
+        neuralEngine: true,
+        neuralEngineTops: 15.8,
+        api: 'Metal 3',
+    },
+    'Apple M2 Ultra': {
+        memory: 192,
+        fp32Tflops: 27.2,
+        fp16Tflops: 54.4,
+        neuralEngine: true,
+        neuralEngineTops: 31.6,
+        api: 'Metal 3',
+    },
+    'Apple M3': {
+        memory: 24,
+        fp32Tflops: 4.0,
+        fp16Tflops: 8.0,
+        neuralEngine: true,
+        neuralEngineTops: 18,
+        api: 'Metal 3',
+    },
+    'Apple M3 Pro': {
+        memory: 36,
+        fp32Tflops: 7.5,
+        fp16Tflops: 15.0,
+        neuralEngine: true,
+        neuralEngineTops: 18,
+        api: 'Metal 3',
+    },
+    'Apple M3 Max': {
+        memory: 128,
+        fp32Tflops: 14.2,
+        fp16Tflops: 28.4,
+        neuralEngine: true,
+        neuralEngineTops: 18,
+        api: 'Metal 3',
+    },
+    'Apple M4': {
+        memory: 32,
+        fp32Tflops: 5.3,
+        fp16Tflops: 10.6,
+        neuralEngine: true,
+        neuralEngineTops: 38,
+        api: 'Metal 3',
+    },
+    'Apple M4 Pro': {
+        memory: 48,
+        fp32Tflops: 12.0,
+        fp16Tflops: 24.0,
+        neuralEngine: true,
+        neuralEngineTops: 38,
+        api: 'Metal 3',
+    },
+    'Apple M4 Max': {
+        memory: 128,
+        fp32Tflops: 18.0,
+        fp16Tflops: 36.0,
+        neuralEngine: true,
+        neuralEngineTops: 38,
+        api: 'Metal 3',
+    },
+};
+/**
+ * Detect all hardware information
+ */
+export async function detectHardware() {
+    const platform = os.platform();
+    const arch = os.arch();
+    const hostname = os.hostname();
+    const cpu = detectCPU();
+    const memory = detectMemory();
+    const gpu = await detectGPU();
+    const frameworks = await detectFrameworks();
+    return {
+        platform,
+        arch,
+        hostname,
+        cpu,
+        memory,
+        gpu,
+        frameworks,
+    };
+}
+/**
+ * Detect CPU information
+ */
+function detectCPU() {
+    const cpus = os.cpus();
+    const model = cpus[0]?.model || 'Unknown';
+    const cores = cpus.length;
+    const speed = cpus[0]?.speed;
+    // Detect Apple Silicon core configuration
+    let performanceCores;
+    let efficiencyCores;
+    if (model.includes('Apple M')) {
+        // Apple Silicon has P-cores and E-cores
+        // M4 Max: 16 cores (12P + 4E)
+        if (model.includes('M4 Max')) {
+            performanceCores = 12;
+            efficiencyCores = 4;
+        }
+        else if (model.includes('M4 Pro')) {
+            performanceCores = 10;
+            efficiencyCores = 4;
+        }
+        else if (model.includes('M4')) {
+            performanceCores = 4;
+            efficiencyCores = 6;
+        }
+        else if (model.includes('M3 Max')) {
+            performanceCores = 12;
+            efficiencyCores = 4;
+        }
+        else if (model.includes('M3 Pro')) {
+            performanceCores = 6;
+            efficiencyCores = 6;
+        }
+        else if (model.includes('M3')) {
+            performanceCores = 4;
+            efficiencyCores = 4;
+        }
+        else {
+            // Generic Apple Silicon estimate
+            performanceCores = Math.ceil(cores * 0.75);
+            efficiencyCores = cores - performanceCores;
+        }
+    }
+    return {
+        model,
+        cores,
+        performanceCores,
+        efficiencyCores,
+        speed,
+    };
+}
+/**
+ * Detect memory information
+ */
+function detectMemory() {
+    const totalBytes = os.totalmem();
+    const freeBytes = os.freemem();
+    const total = Math.round(totalBytes / (1024 * 1024 * 1024));
+    const available = Math.round(freeBytes / (1024 * 1024 * 1024));
+    // Apple Silicon uses unified memory
+    const cpus = os.cpus();
+    const unified = cpus[0]?.model.includes('Apple M') || false;
+    return {
+        total,
+        available,
+        unified,
+    };
+}
+/**
+ * Detect GPU/Accelerator
+ */
+async function detectGPU() {
+    const platform = os.platform();
+    const cpus = os.cpus();
+    const cpuModel = cpus[0]?.model || '';
+    // Check for Apple Silicon
+    if (cpuModel.includes('Apple M')) {
+        // Find matching Apple Silicon specs
+        for (const [model, specs] of Object.entries(APPLE_SILICON_SPECS)) {
+            if (cpuModel.includes(model.replace('Apple ', ''))) {
+                const totalMemory = Math.round(os.totalmem() / (1024 * 1024 * 1024));
+                return {
+                    type: model,
+                    memory: totalMemory, // Unified memory
+                    api: specs.api || 'Metal 3',
+                    fp32Tflops: specs.fp32Tflops,
+                    fp16Tflops: specs.fp16Tflops,
+                    computeCapability: 'metal-3',
+                    neuralEngine: specs.neuralEngine,
+                    neuralEngineTops: specs.neuralEngineTops,
+                };
+            }
+        }
+        // Generic Apple Silicon
+        const totalMemory = Math.round(os.totalmem() / (1024 * 1024 * 1024));
+        return {
+            type: cpuModel,
+            memory: totalMemory,
+            api: 'Metal 3',
+            computeCapability: 'metal-3',
+        };
+    }
+    // Check for NVIDIA GPU on Linux/Windows
+    if (platform === 'linux' || platform === 'win32') {
+        try {
+            const result = execSync('nvidia-smi --query-gpu=name,memory.total --format=csv,noheader', {
+                encoding: 'utf-8',
+                stdio: ['pipe', 'pipe', 'ignore'],
+            });
+            const lines = result.trim().split('\n');
+            if (lines.length > 0) {
+                const [name, memoryStr] = lines[0].split(',').map((s) => s.trim());
+                const memory = parseInt(memoryStr) / 1024; // Convert MiB to GB
+                return {
+                    type: name,
+                    memory: Math.round(memory),
+                    api: 'CUDA',
+                    computeCapability: await detectCudaCapability(),
+                };
+            }
+        }
+        catch {
+            // nvidia-smi not available or no GPU
+        }
+    }
+    return null;
+}
+/**
+ * Detect CUDA compute capability
+ */
+async function detectCudaCapability() {
+    try {
+        const result = execSync('nvidia-smi --query-gpu=compute_cap --format=csv,noheader', { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'ignore'] });
+        return result.trim();
+    }
+    catch {
+        return undefined;
+    }
+}
+/**
+ * Detect installed ML frameworks
+ */
+async function detectFrameworks() {
+    const frameworks = [];
+    const platform = os.platform();
+    const cpus = os.cpus();
+    const isAppleSilicon = cpus[0]?.model.includes('Apple M');
+    // Check PyTorch
+    try {
+        const result = execSync('python3 -c "import torch; print(torch.__version__)"', {
+            encoding: 'utf-8',
+            stdio: ['pipe', 'pipe', 'ignore'],
+        });
+        const version = result.trim();
+        // Check for GPU support
+        let gpuSupport = false;
+        try {
+            if (isAppleSilicon) {
+                const mpsCheck = execSync('python3 -c "import torch; print(torch.backends.mps.is_available())"', {
+                    encoding: 'utf-8',
+                    stdio: ['pipe', 'pipe', 'ignore'],
+                });
+                gpuSupport = mpsCheck.trim() === 'True';
+            }
+            else {
+                const cudaCheck = execSync('python3 -c "import torch; print(torch.cuda.is_available())"', {
+                    encoding: 'utf-8',
+                    stdio: ['pipe', 'pipe', 'ignore'],
+                });
+                gpuSupport = cudaCheck.trim() === 'True';
+            }
+        }
+        catch {
+            // GPU check failed
+        }
+        frameworks.push({
+            name: 'PyTorch',
+            version,
+            available: true,
+            gpuSupport,
+        });
+    }
+    catch {
+        frameworks.push({
+            name: 'PyTorch',
+            available: false,
+            gpuSupport: false,
+        });
+    }
+    // Check TensorFlow
+    try {
+        const result = execSync('python3 -c "import tensorflow as tf; print(tf.__version__)"', {
+            encoding: 'utf-8',
+            stdio: ['pipe', 'pipe', 'ignore'],
+        });
+        const version = result.trim();
+        // Check for GPU support
+        let gpuSupport = false;
+        try {
+            const gpuCheck = execSync('python3 -c "import tensorflow as tf; print(len(tf.config.list_physical_devices(\'GPU\')) > 0)"', {
+                encoding: 'utf-8',
+                stdio: ['pipe', 'pipe', 'ignore'],
+            });
+            gpuSupport = gpuCheck.trim() === 'True';
+        }
+        catch {
+            // GPU check failed
+        }
+        frameworks.push({
+            name: 'TensorFlow',
+            version,
+            available: true,
+            gpuSupport,
+        });
+    }
+    catch {
+        frameworks.push({
+            name: 'TensorFlow',
+            available: false,
+            gpuSupport: false,
+        });
+    }
+    // Check MLX (Apple Silicon only)
+    if (isAppleSilicon) {
+        try {
+            const result = execSync('python3 -c "import mlx; print(mlx.__version__)"', {
+                encoding: 'utf-8',
+                stdio: ['pipe', 'pipe', 'ignore'],
+            });
+            const version = result.trim();
+            frameworks.push({
+                name: 'MLX',
+                version,
+                available: true,
+                gpuSupport: true, // MLX always uses GPU on Apple Silicon
+            });
+        }
+        catch {
+            frameworks.push({
+                name: 'MLX',
+                available: false,
+                gpuSupport: false,
+            });
+        }
+    }
+    // Check JAX
+    try {
+        const result = execSync('python3 -c "import jax; print(jax.__version__)"', {
+            encoding: 'utf-8',
+            stdio: ['pipe', 'pipe', 'ignore'],
+        });
+        const version = result.trim();
+        // Check for GPU support
+        let gpuSupport = false;
+        try {
+            const gpuCheck = execSync('python3 -c "import jax; print(len(jax.devices(\'gpu\')) > 0)"', {
+                encoding: 'utf-8',
+                stdio: ['pipe', 'pipe', 'ignore'],
+            });
+            gpuSupport = gpuCheck.trim() === 'True';
+        }
+        catch {
+            // GPU check failed
+        }
+        frameworks.push({
+            name: 'JAX',
+            version,
+            available: true,
+            gpuSupport,
+        });
+    }
+    catch {
+        frameworks.push({
+            name: 'JAX',
+            available: false,
+            gpuSupport: false,
+        });
+    }
+    return frameworks;
+}
+export default detectHardware;
+//# sourceMappingURL=hardware-detection.js.map

package/dist/commands/compute/lib/hardware-detection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hardware-detection.js","sourceRoot":"","sources":["../../../../src/commands/compute/lib/hardware-detection.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AA4CzC;;GAEG;AACH,MAAM,mBAAmB,GAAqC;IAC5D,UAAU,EAAE;QACV,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,GAAG;QACf,UAAU,EAAE,GAAG;QACf,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,EAAE;QACpB,GAAG,EAAE,SAAS;KACf;IACD,cAAc,EAAE;QACd,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,GAAG;QACf,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,EAAE;QACpB,GAAG,EAAE,SAAS;KACf;IACD,cAAc,EAAE;QACd,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,EAAE;QACpB,GAAG,EAAE,SAAS;KACf;IACD,gBAAgB,EAAE;QAChB,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,EAAE;QACpB,GAAG,EAAE,SAAS;KACf;IACD,UAAU,EAAE;QACV,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,GAAG;QACf,UAAU,EAAE,GAAG;QACf,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,IAAI;QACtB,GAAG,EAAE,SAAS;KACf;IACD,cAAc,EAAE;QACd,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,GAAG;QACf,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,IAAI;QACtB,GAAG,EAAE,SAAS;KACf;IACD,cAAc,EAAE;QACd,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,IAAI;QACtB,GAAG,EAAE,SAAS;KACf;IACD,gBAAgB,EAAE;QAChB,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,IAAI;QACtB,GAAG,EAAE,SAAS;KACf;IACD,UAAU,EAAE;QACV,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,GAAG;QACf,UAAU,EAAE,GAAG;QACf,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,EAAE;QACpB,GAAG,EAAE,SAAS;KACf;IACD,cAAc,EAAE;QACd,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,GAAG;QACf,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,EAAE;QACpB,GAAG,EAAE,SAAS;KACf;IACD,cAAc,EAAE;QACd,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,EAAE;QACpB,GAAG,EAAE,SAAS;KACf;IACD,UAAU,EAAE;QACV,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,GAAG;QACf,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,EAAE;QACpB,GAAG,EAAE,SAAS;KACf;IACD,cAAc,EAAE;QACd,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,EAAE;QACpB,GAAG,EAAE,SAAS;KACf;IACD,cAAc,EAAE;QACd,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,EAAE;QACpB,GAAG,EAAE,SAAS;KACf;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc;IAClC,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;IAC/B,MAAM,IAAI,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC;IACvB,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;IAE/B,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;IACxB,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;IAC9B,MAAM,GAAG,GAAG,MAAM,SAAS,EAAE,CAAC;IAC9B,MAAM,UAAU,GAAG,MAAM,gBAAgB,EAAE,CAAC;IAE5C,OAAO;QACL,QAAQ;QACR,IAAI;QACJ,QAAQ;QACR,GAAG;QACH,MAAM;QACN,GAAG;QACH,UAAU;KACX,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,SAAS;IAChB,MAAM,IAAI,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC;IACvB,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,SAAS,CAAC;IAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC;IAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;IAE7B,0CAA0C;IAC1C,IAAI,gBAAoC,CAAC;IACzC,IAAI,eAAmC,CAAC;IAExC,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAC9B,wCAAwC;QACxC,8BAA8B;QAC9B,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7B,gBAAgB,GAAG,EAAE,CAAC;YACtB,eAAe,GAAG,CAAC,CAAC;QACtB,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACpC,gBAAgB,GAAG,EAAE,CAAC;YACtB,eAAe,GAAG,CAAC,CAAC;QACtB,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YAChC,gBAAgB,GAAG,CAAC,CAAC;YACrB,eAAe,GAAG,CAAC,CAAC;QACtB,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACpC,gBAAgB,GAAG,EAAE,CAAC;YACtB,eAAe,GAAG,CAAC,CAAC;QACtB,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACpC,gBAAgB,GAAG,CAAC,CAAC;YACrB,eAAe,GAAG,CAAC,CAAC;QACtB,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YAChC,gBAAgB,GAAG,CAAC,CAAC;YACrB,eAAe,GAAG,CAAC,CAAC;QACtB,CAAC;aAAM,CAAC;YACN,iCAAiC;YACjC,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC;YAC3C,eAAe,GAAG,KAAK,GAAG,gBAAgB,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,OAAO;QACL,KAAK;QACL,KAAK;QACL,gBAAgB;QAChB,eAAe;QACf,KAAK;KACN,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,YAAY;IACnB,MAAM,UAAU,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;IACjC,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;IAE/B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC;IAC5D,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC;IAE/D,oCAAoC;IACpC,MAAM,IAAI,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC;IACvB,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC;IAE5D,OAAO;QACL,KAAK;QACL,SAAS;QACT,OAAO;KACR,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,SAAS;IACtB,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;IAC/B,MAAM,IAAI,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC;IACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,EAAE,CAAC;IAEtC,0BAA0B;IAC1B,IAAI,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACjC,oCAAoC;QACpC,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,EAAE,CAAC;YACjE,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC;gBACnD,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC;gBACrE,OAAO;oBACL,IAAI,EAAE,KAAK;oBACX,MAAM,EAAE,WAAW,EAAE,iBAAiB;oBACtC,GAAG,EAAE,KAAK,CAAC,GAAG,IAAI,SAAS;oBAC3B,UAAU,EAAE,KAAK,CAAC,UAAU;oBAC5B,UAAU,EAAE,KAAK,CAAC,UAAU;oBAC5B,iBAAiB,EAAE,SAAS;oBAC5B,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;iBACzC,CAAC;YACJ,CAAC;QACH,CAAC;QAED,wBAAwB;QACxB,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC;QACrE,OAAO;YACL,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,WAAW;YACnB,GAAG,EAAE,SAAS;YACd,iBAAiB,EAAE,SAAS;SAC7B,CAAC;IACJ,CAAC;IAED,wCAAwC;IACxC,IAAI,QAAQ,KAAK,OAAO,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QACjD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,QAAQ,CAAC,gEAAgE,EAAE;gBACxF,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;aAClC,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACxC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrB,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;gBACnE,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,oBAAoB;gBAE/D,OAAO;oBACL,IAAI,EAAE,IAAI;oBACV,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;oBAC1B,GAAG,EAAE,MAAM;oBACX,iBAAiB,EAAE,MAAM,oBAAoB,EAAE;iBAChD,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,qCAAqC;QACvC,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,oBAAoB;IACjC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,QAAQ,CACrB,0DAA0D,EAC1D,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,CACzD,CAAC;QACF,OAAO,MAAM,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,gBAAgB;IAC7B,MAAM,UAAU,GAAoB,EAAE,CAAC;IACvC,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;IAC/B,MAAM,IAAI,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC;IACvB,MAAM,cAAc,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAE1D,gBAAgB;IAChB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,QAAQ,CAAC,qDAAqD,EAAE;YAC7E,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;SAClC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;QAE9B,wBAAwB;QACxB,IAAI,UAAU,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC;YACH,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,QAAQ,GAAG,QAAQ,CAAC,qEAAqE,EAAE;oBAC/F,QAAQ,EAAE,OAAO;oBACjB,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;iBAClC,CAAC,CAAC;gBACH,UAAU,GAAG,QAAQ,CAAC,IAAI,EAAE,KAAK,MAAM,CAAC;YAC1C,CAAC;iBAAM,CAAC;gBACN,MAAM,SAAS,GAAG,QAAQ,CAAC,6DAA6D,EAAE;oBACxF,QAAQ,EAAE,OAAO;oBACjB,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;iBAClC,CAAC,CAAC;gBACH,UAAU,GAAG,SAAS,CAAC,IAAI,EAAE,KAAK,MAAM,CAAC;YAC3C,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,mBAAmB;QACrB,CAAC;QAED,UAAU,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,SAAS;YACf,OAAO;YACP,SAAS,EAAE,IAAI;YACf,UAAU;SACX,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,UAAU,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,SAAS;YACf,SAAS,EAAE,KAAK;YAChB,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;IACL,CAAC;IAED,mBAAmB;IACnB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,QAAQ,CAAC,6DAA6D,EAAE;YACrF,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;SAClC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;QAE9B,wBAAwB;QACxB,IAAI,UAAU,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,QAAQ,CAAC,gGAAgG,EAAE;gBAC1H,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;aAClC,CAAC,CAAC;YACH,UAAU,GAAG,QAAQ,CAAC,IAAI,EAAE,KAAK,MAAM,CAAC;QAC1C,CAAC;QAAC,MAAM,CAAC;YACP,mBAAmB;QACrB,CAAC;QAED,UAAU,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,YAAY;YAClB,OAAO;YACP,SAAS,EAAE,IAAI;YACf,UAAU;SACX,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,UAAU,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,YAAY;YAClB,SAAS,EAAE,KAAK;YAChB,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;IACL,CAAC;IAED,iCAAiC;IACjC,IAAI,cAAc,EAAE,CAAC;QACnB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,QAAQ,CAAC,iDAAiD,EAAE;gBACzE,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;aAClC,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;YAE9B,UAAU,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,KAAK;gBACX,OAAO;gBACP,SAAS,EAAE,IAAI;gBACf,UAAU,EAAE,IAAI,EAAE,uCAAuC;aAC1D,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,UAAU,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,KAAK;gBACX,SAAS,EAAE,KAAK;gBAChB,UAAU,EAAE,KAAK;aAClB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,YAAY;IACZ,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,QAAQ,CAAC,iDAAiD,EAAE;YACzE,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;SAClC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;QAE9B,wBAAwB;QACxB,IAAI,UAAU,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,QAAQ,CAAC,+DAA+D,EAAE;gBACzF,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;aAClC,CAAC,CAAC;YACH,UAAU,GAAG,QAAQ,CAAC,IAAI,EAAE,KAAK,MAAM,CAAC;QAC1C,CAAC;QAAC,MAAM,CAAC;YACP,mBAAmB;QACrB,CAAC;QAED,UAAU,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,KAAK;YACX,OAAO;YACP,SAAS,EAAE,IAAI;YACf,UAAU;SACX,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,UAAU,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,KAAK;YACX,SAAS,EAAE,KAAK;YAChB,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,eAAe,cAAc,CAAC"}

package/dist/commands/compute/lib/local-compute-agent.d.ts

@@ -0,0 +1,143 @@
+/**
+ * Local Compute Agent
+ *
+ * Daemon process that registers with HPC Gateway and executes ML jobs locally.
+ * Supports Apple Silicon (MPS), NVIDIA CUDA, and CPU-only execution.
+ */
+import EventEmitter from 'eventemitter3';
+import type { LocalComputeConfig, ComputeAgentStatus, LocalAgentStatus, LocalComputeJob, JobStatus, ComputeEvents, KernelSession, ExecuteRequest, ExecuteResult } from '../../../types/compute.js';
+export declare class LocalComputeAgent extends EventEmitter<ComputeEvents> {
+    private config;
+    private hardware;
+    private agentId;
+    private gatewaySocket;
+    private localServer;
+    private localApp;
+    private localIO;
+    private jobQueue;
+    private currentJob;
+    private completedJobs;
+    private jobLogs;
+    private kernelSessions;
+    private kernelProcesses;
+    private kernelOutputBuffers;
+    private reconnectAttempts;
+    private reconnectTimer;
+    private heartbeatTimer;
+    private idleTimer;
+    private running;
+    private jobsCompleted;
+    private jobsFailed;
+    private totalComputeTime;
+    private startTime;
+    constructor(config?: Partial<LocalComputeConfig>);
+    /**
+     * Start the local compute agent
+     */
+    start(): Promise<void>;
+    /**
+     * Stop the agent gracefully
+     */
+    stop(): Promise<void>;
+    /**
+     * Submit a job for execution
+     */
+    submitJob(job: Omit<LocalComputeJob, 'id' | 'status' | 'submittedAt' | 'logs'>): Promise<LocalComputeJob>;
+    /**
+     * Get job by ID
+     */
+    getJob(jobId: string): LocalComputeJob | null;
+    /**
+     * List all jobs
+     */
+    listJobs(options?: {
+        status?: JobStatus;
+        limit?: number;
+    }): LocalComputeJob[];
+    /**
+     * Cancel a job
+     */
+    cancelJob(jobId: string): Promise<boolean>;
+    /**
+     * Get job logs
+     */
+    getJobLogs(jobId: string, tail?: number): string[];
+    /**
+     * Get agent status for local CLI display
+     */
+    getStatus(): Promise<LocalAgentStatus>;
+    /**
+     * Get full agent status for HPC Gateway
+     */
+    getFullStatus(): ComputeAgentStatus;
+    /**
+     * Start agent as a background daemon
+     */
+    startDaemon(): Promise<void>;
+    /**
+     * Check if another agent is already running
+     */
+    private isAnotherAgentRunning;
+    /**
+     * Start local API server
+     */
+    private startLocalServer;
+    /**
+     * Connect to HPC Gateway
+     */
+    private connectToGateway;
+    /**
+     * Schedule reconnection attempt
+     */
+    private scheduleReconnect;
+    /**
+     * Start heartbeat to gateway
+     */
+    private startHeartbeat;
+    /**
+     * Reset idle timer
+     */
+    private resetIdleTimer;
+    /**
+     * Process job queue
+     */
+    private processQueue;
+    /**
+     * Execute a job
+     */
+    private executeJob;
+    /**
+     * Create a new kernel session
+     */
+    createKernel(language?: 'python' | 'r'): Promise<KernelSession>;
+    /**
+     * Execute code in a kernel
+     */
+    executeCode(request: ExecuteRequest): Promise<ExecuteResult>;
+    /**
+     * Shutdown a kernel
+     */
+    shutdownKernel(kernelId: string): Promise<void>;
+    /**
+     * Get kernel session info
+     */
+    getKernelSession(kernelId: string): KernelSession | null;
+    /**
+     * List all active kernels
+     */
+    listKernels(): KernelSession[];
+    /**
+     * Interrupt kernel execution
+     */
+    interruptKernel(kernelId: string): boolean;
+    /**
+     * Shutdown all kernels (called on agent stop)
+     */
+    private shutdownAllKernels;
+    /**
+     * Setup graceful shutdown handlers
+     */
+    private setupShutdownHandlers;
+}
+export default LocalComputeAgent;
+//# sourceMappingURL=local-compute-agent.d.ts.map

package/dist/commands/compute/lib/local-compute-agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"local-compute-agent.d.ts","sourceRoot":"","sources":["../../../../src/commands/compute/lib/local-compute-agent.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAOH,OAAO,YAAY,MAAM,eAAe,CAAC;AAMzC,OAAO,KAAK,EACV,kBAAkB,EAElB,kBAAkB,EAClB,gBAAgB,EAChB,eAAe,EAEf,SAAS,EAET,aAAa,EAGb,aAAa,EAEb,cAAc,EACd,aAAa,EAEd,MAAM,2BAA2B,CAAC;AASnC,qBAAa,iBAAkB,SAAQ,YAAY,CAAC,aAAa,CAAC;IAChE,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,QAAQ,CAA6B;IAC7C,OAAO,CAAC,OAAO,CAAuB;IACtC,OAAO,CAAC,aAAa,CAAuB;IAC5C,OAAO,CAAC,WAAW,CAAuB;IAC1C,OAAO,CAAC,QAAQ,CAAwB;IACxC,OAAO,CAAC,OAAO,CAA+B;IAE9C,OAAO,CAAC,QAAQ,CAAmB;IACnC,OAAO,CAAC,UAAU,CAA2B;IAC7C,OAAO,CAAC,aAAa,CAA2C;IAChE,OAAO,CAAC,OAAO,CAAoC;IAGnD,OAAO,CAAC,cAAc,CAAyC;IAC/D,OAAO,CAAC,eAAe,CAAwC;IAC/D,OAAO,CAAC,mBAAmB,CAA6C;IAExE,OAAO,CAAC,iBAAiB,CAAa;IACtC,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,SAAS,CAA+B;IAChD,OAAO,CAAC,OAAO,CAAkB;IAEjC,OAAO,CAAC,aAAa,CAAa;IAClC,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,gBAAgB,CAAa;IACrC,OAAO,CAAC,SAAS,CAAqB;gBAE1B,MAAM,GAAE,OAAO,CAAC,kBAAkB,CAAM;IAepD;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IA0C5B;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAsD3B;;OAEG;IACG,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,eAAe,EAAE,IAAI,GAAG,QAAQ,GAAG,aAAa,GAAG,MAAM,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC;IA2B/G;;OAEG;IACH,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,eAAe,GAAG,IAAI;IAgB7C;;OAEG;IACH,QAAQ,CAAC,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,SAAS,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,eAAe,EAAE;IA8B7E;;OAEG;IACG,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAmDhD;;OAEG;IACH,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE;IAQlD;;OAEG;IACG,SAAS,IAAI,OAAO,CAAC,gBAAgB,CAAC;IAgC5C;;OAEG;IACH,aAAa,IAAI,kBAAkB;IAenC;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;IAiBlC;;OAEG;YACW,qBAAqB;IAuBnC;;OAEG;YACW,gBAAgB;IAiM9B;;OAEG;YACW,gBAAgB;IAyF9B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IA0BzB;;OAEG;IACH,OAAO,CAAC,cAAc;IAetB;;OAEG;IACH,OAAO,CAAC,cAAc;IAatB;;OAEG;YACW,YAAY;IAkB1B;;OAEG;YACW,UAAU;IA6JxB;;OAEG;IACG,YAAY,CAAC,QAAQ,GAAE,QAAQ,GAAG,GAAc,GAAG,OAAO,CAAC,aAAa,CAAC;IA0G/E;;OAEG;IACG,WAAW,CAAC,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,aAAa,CAAC;IAsLlE;;OAEG;IACG,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAkCrD;;OAEG;IACH,gBAAgB,CAAC,QAAQ,EAAE,MAAM,GAAG,aAAa,GAAG,IAAI;IAIxD;;OAEG;IACH,WAAW,IAAI,aAAa,EAAE;IAI9B;;OAEG;IACH,eAAe,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAqB1C;;OAEG;YACW,kBAAkB;IAOhC;;OAEG;IACH,OAAO,CAAC,qBAAqB;CAU9B;AAED,eAAe,iBAAiB,CAAC"}