CompiledCudaComplexEvaluator

CompiledCudaComplexEvaluator

CompiledCudaComplexEvaluator()

A compiled evaluator of an expression that uses CUDA for GPU acceleration.

Methods

Name Description
evaluate Evaluate the expression for multiple inputs and return the result.
load Load a compiled library, previously generated with Evaluator.compile().

evaluate

CompiledCudaComplexEvaluator.evaluate(inputs: npt.ArrayLike) -> npt.NDArray[np.complex128]

Evaluate the expression for multiple inputs and return the result.

load

CompiledCudaComplexEvaluator.load(
    filename: str,
    function_name: str,
    input_len: int,
    output_len: int,
    cuda_number_of_evaluations: int,
    cuda_block_size: Optional[int] = 256,
) -> CompiledCudaComplexEvaluator

Load a compiled library, previously generated with Evaluator.compile().