feat(cpp/datascience): GPU Monte Carlo kernels (K1-K3)

Tres kernels Monte Carlo intensivos sobre las primitivas G1-G7 + las puras
CPU como oraculo de tests numericos. Apuntados a hyper-paralelizar los
calculadores de sources/calculadoras (vr_tiered_lab, mcmc-bayes / full / lab,
mcmc-visualizer) en RTX-class GPUs.

- mc_session_sim_gpu (K1): N sesiones independientes de K spins en paralelo
  (1 thread = 1 sesion). Modelo variable-ratio escalonado con tiers (q, m),
  modes Pure/Pity/Streak, miss_streak, drawdown. SSBOs summary[N*8] y
  tier_counts[N*max_tiers]. Portea vr_tiered_lab.
- mc_metropolis_hastings_gpu (K2): M cadenas independientes 1D. Target
  log-pdf inyectable como string GLSL (mismo patron de gl_shader). u_user[16]
  para cambiar parametros desde sliders sin recompilar. Output compatible
  con rhat_split / ess_basic.
- mc_random_walk_2d_gpu (K3): walkers 2D MH con trace_xy xy-interleaved en
  SSBO; pasable directamente a gpu_histogram_2d sin readback intermedio.
  Pipeline GPU-only para mcmc-visualizer.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
2026-05-04 11:52:41 +02:00
parent d115d8e830
commit 7b0384c804
9 changed files with 1123 additions and 0 deletions
@@ -0,0 +1,90 @@
---
name: mc_metropolis_hastings_gpu
kind: function
lang: cpp
domain: datascience
version: "1.0.0"
purity: impure
signature: "McMetropolisHastingsGpu mc_mh_gpu_create(int m_chains, int n_samples_per_run, const std::string& target_log_pdf_glsl); void mc_mh_gpu_reset(...); void mc_mh_gpu_run(...); void mc_mh_gpu_readback_chains(...); void mc_mh_gpu_readback_accepts(...); void mc_mh_gpu_destroy(...)"
description: "Metropolis-Hastings 1D paralelo en GPU: M cadenas independientes (1 thread = 1 chain). Target log-pdf inyectable como string GLSL (igual patron que gl_shader). Soporta u_user[16] para parametros sin recompilar."
tags: [montecarlo, mcmc, metropolis, gpu, sampling, datascience]
uses_functions: ["gl_loader_cpp_gfx", "gpu_ssbo_cpp_gfx", "gpu_compute_program_cpp_gfx", "gpu_dispatch_cpp_gfx", "gpu_rng_glsl_cpp_gfx"]
uses_types: []
returns: []
returns_optional: false
error_type: "error_go_core"
imports: [GL/gl.h, GL/glext.h, vector, string, cstdio]
tested: false
tests: []
test_file_path: ""
file_path: "cpp/functions/datascience/mc_metropolis_hastings_gpu.cpp"
framework: opengl
params:
- name: m_chains
desc: "Numero de cadenas independientes (1 thread por cadena). Tipico 4-32."
- name: n_samples_per_run
desc: "Steps por cadena en cada llamada a run. Subsiguientes runs continuan; chains[] se sobreescribe pero curr_x persiste."
- name: target_log_pdf_glsl
desc: "Snippet GLSL que define float target_log_pdf(float x). Puede leer u_user[0..15]. NO normaliza necesariamente."
- name: master_seed
desc: "Semilla base para SplitMix64. Cada cadena recibe un PCG state distinto."
- name: x0
desc: "Valor inicial. Mismo para todas las cadenas si initial_xs == nullptr."
- name: initial_xs
desc: "(opcional) array m_chains floats. Useful para cadenas dispersas que cubran todos los modos."
- name: proposal_sigma
desc: "Stddev del proposal Gaussian. Tunear hasta accept rate ~0.2-0.4."
- name: user_params
desc: "(opcional) array <= 16 floats. Se sube a u_user[]."
output: "chains[m * n_samples_per_run] floats row-major (chain j, sample i = out[j*n + i]) — compatible con rhat_split / ess_basic. accept_counts[m] uints — accept_rate = counts[j] / total_steps_acumulado."
---
# mc_metropolis_hastings_gpu
Equivalente GPU de `metropolis_hastings_cpp_datascience`. La diferencia clave: la log-pdf se inyecta como string GLSL (no `std::function`) y se evalua dentro del kernel — sin call overhead, todo el inner loop ocurre en GPU.
## Patron tipico (mcmc-bayes Beta-Binomial)
```cpp
const std::string log_pdf = R"glsl(
float target_log_pdf(float theta) {
if (theta <= 0.0 || theta >= 1.0) return -1e30;
float a = u_user[0]; // alpha + k
float b = u_user[1]; // beta + (n - k)
return (a - 1.0) * log(theta) + (b - 1.0) * log(1.0 - theta);
}
)glsl";
auto mh = fn::ds::mc_mh_gpu_create(/*m_chains=*/8,
/*n_samples_per_run=*/100'000,
log_pdf);
fn::ds::mc_mh_gpu_reset(mh, /*seed=*/0xC0FFEE, /*x0=*/0.5f);
float user[2] = { 8.0f, 4.0f }; // alpha+k, beta+(n-k)
fn::ds::mc_mh_gpu_run(mh, /*proposal_sigma=*/0.1f, user, 2);
std::vector<float> chains(8 * 100'000);
fn::ds::mc_mh_gpu_readback_chains(mh, chains.data());
// Convergencia: convertir a double y pasar a rhat_split
std::vector<double> chains_d(chains.begin(), chains.end());
double r = fn::ds::rhat_split(chains_d.data(), 8, 100'000);
fn::ds::mc_mh_gpu_destroy(mh);
```
## Performance
RTX 3070, 8 cadenas × 10^6 steps con log-pdf simple: **~50 ms total**. Comparado con CPU single-thread (~3 s) son 60x. La ganancia se nota cuando arrastras un slider y quieres convergencia visible en cada frame.
## u_user[]
Array de 16 floats que el shader puede leer sin recompilar. Asi puedes cambiar `alpha`, `beta`, `mu_prior`, `sigma_prior` etc. desde sliders sin volver a llamar `create`. Si necesitas mas de 16 parametros: agrupar en SSBO custom.
## Notas
- **chains[] se sobreescribe en cada run**. Si llamas `run` dos veces sin leer, solo conservas la cadena del segundo run. Para muestrear largos en chunks: leer entre runs.
- **curr_x persiste**: el sampler continua de donde estaba. Solo `reset` rebobina al x0.
- **accept_counts se acumulan**: para tener accept_rate por run individual, lee antes y despues y resta.
- target_log_pdf devuelve `float` — fp32 es suficiente para MC tipico. Para log-likelihoods muy condicionados (donde la diferencia entre log_p_curr y log_p_prop puede ser >10^7) considerar log-sum-exp manual o saturar antes de exp().
- Para sample 2D / d-dim: extender el body a `float[d] target_log_pdf(float[d] x)` y arrays de proposal_sigma. No incluido en esta version (usar mc_random_walk_2d_gpu para 2D especificamente).