Skip to main content

Module gpu_weights

Module gpu_weights 

Source
Expand description

GPU-native weight table builders for exact inference.

Structs§

GpuWeights
Device-resident log weights indexed by DIMACS variable.

Functions§

apply_query_vars_device
Temporarily forces query variables true by replacing their false log weights.
build_evidence_by_var_gpu
Builds a device table that maps DIMACS variables to forced evidence values.
build_weights_gpu
Builds validated device log-weight tables from probabilities and evidence.
map_nodes_to_vars_gpu
Maps device-resident PIR node identifiers to their DIMACS variables.
restore_query_vars_device
Restores false log weights saved by apply_query_vars_device.