Method _create_test_program

rightnow_cli/profiler.py:204–247

Create a test program that runs the kernel.

(
        self,
        compiled_kernel: Dict[str, Any],
        model_info: Dict[str, Any],
        temp_path: Path
    )

                return self._estimate_metrics(compiled_kernel, model_info)
    
    def _create_test_program(
        self,
        compiled_kernel: Dict[str, Any],
        model_info: Dict[str, Any],
        temp_path: Path
    ) -> Optional[Path]:
        """Create a test program that runs the kernel."""
        operation = compiled_kernel.get("operation", "unknown")
        kernel_code = compiled_kernel.get("full_code", "")
        
        if not kernel_code:
            return None
        
        test_code = self._generate_test_harness(operation, kernel_code, model_info)
        
        cuda_file = temp_path / "test_kernel.cu"
        exe_file = temp_path / ("test_kernel.exe" if platform.system() == "Windows" else "test_kernel")
        
        with open(cuda_file, 'w') as f:
            f.write(test_code)
        
        from .compiler import CUDACompiler
        compiler = CUDACompiler()
        
        compile_cmd = [
            compiler.nvcc_path,
            "-O3",
            "-std=c++14"
        ] + compiler.arch_flags + [
            "-o", str(exe_file),
            str(cuda_file)
        ]
        
        try:
            result = subprocess.run(
                compile_cmd,
                capture_output=True,
                text=True,
                check=True
            )
            return exe_file
        except subprocess.CalledProcessError as e:
            console.print(f"[red]Failed to compile test program: {e.stderr}[/red]")
            return None
    
    def _generate_test_harness(
        self,