omnisolver.bruteforce.gpu#

Implementation of parallel tempering plugin for Omnisolver.

Submodules#

Package Contents#

Classes#

BruteforceGPUSampler

Functions#

get_plugin(→ omnisolver.common.plugin.Plugin)

Get package name and resource path.
class omnisolver.bruteforce.gpu.BruteforceGPUSampler#

Bases: dimod.Sampler

property parameters: Dict[str, Any]#
property properties: Dict[str, Any]#
sample(bqm, num_states, suffix_size, grid_size, block_size, dtype=np.float32)#

Solve Binary Quadratic Model using exhaustive (bruteforce) search on the GPU.

Parameters:

  • bqm – Binary Quadratic Model instance to solve.

  • num_states – number of lowest energy states to compute.

  • suffix_size – exponent l such that 2 ** l is the number of temporarily stored

  • grid_size – number of blocks for the custom kernels. Note that this parameter does not affect the grid on which the Thrust kernels are launched.

  • block_size – number of threads per block for custom kerneles. Note that this parameter does not affect the grid on which the Thrust kernels are launched.

  • dtype – datatype to use, either np.float32 or np.float64. The default is np.flaot32 which on most GPU is significantly faster then 64-bit floating point numbers.

Returns:

sample set containing num_states samples.

omnisolver.bruteforce.gpu.get_plugin() → omnisolver.common.plugin.Plugin#

Get package name and resource path.