Method _run_optimization

rightnow_cli/interactive.py:335–482

Run optimization - SIMPLIFIED.

(self, file: Path, kernels: List[Dict], options: Dict)

        }

    def _run_optimization(self, file: Path, kernels: List[Dict], options: Dict):
        """Run optimization - SIMPLIFIED."""

        from .kernel_analyzer import KernelAnalyzer
        from .openrouter import OpenRouterClient
        from .compiler import CUDACompiler
        from .bench import Benchmarker

        api_key = self.cache_manager.get_api_key()
        client = OpenRouterClient(api_key)
        analyzer = KernelAnalyzer()
        compiler = CUDACompiler()
        benchmarker = Benchmarker(iterations=options['iterations'])

        for idx, kernel in enumerate(kernels, 1):
            console.print(f"\n[bold cyan]Kernel {idx}/{len(kernels)}: {kernel['name']}[/bold cyan]")

            # Analyze
            console.print("[dim]Analyzing...[/dim]")
            try:
                analysis = analyzer.analyze_kernel(kernel['code'])
                console.print(f"[green]✓ Found {len(analysis.get('optimization_opportunities', []))} optimization opportunities[/green]")
            except Exception as e:
                console.print(f"[red]✗ Analysis failed: {e}[/red]")
                continue

            # Generate variants
            console.print(f"[dim]Generating {options['variants']} variants with AI...[/dim]")

            constraints = {
                'max_registers': options['max_registers'],
                'shared_memory_kb': options['shared_mem_kb'],
                'target_gpu': options['target_gpu']
            }

            variants = []
            with Progress(
                SpinnerColumn(),
                TextColumn("[dim]{task.description}[/dim]"),
                BarColumn(bar_width=30),
                TextColumn("{task.percentage:>3.0f}%"),
                console=console,
                transient=True
            ) as progress:
                task = progress.add_task("", total=options['variants'])

                for i in range(options['variants']):
                    try:
                        generated = client.generate_kernel_optimizations(
                            original_code=kernel['code'],
                            analysis=analysis,
                            constraints=constraints,
                            num_variants=1
                        )
                        if generated:
                            variants.extend(generated)
                    except Exception as e:
                        console.print(f"[yellow]Variant {i+1} failed: {str(e)[:50]}[/yellow]")