Template Function Pennylane::LightningGPU::Util::scaleC_CUDA

• Defined in File LinearAlg.hpp

Function Documentation

template<class CFP_t = std::complex<double>, class T = cuDoubleComplex, class DevTypeID = int>
inline auto Pennylane::LightningGPU::Util::scaleC_CUDA(const CFP_t a, T *v1, const int data_size, DevTypeID dev_id, cudaStream_t stream_id, const CublasCaller &cublas)

cuBLAS backed GPU data scaling.

Template Parameters

CFP_t – Complex data-type. Accepts std::complex<float> and std::complex<double>

Parameters

• a – scaling factor

• v1 – Device data pointer

• data_size – Length of device data.

• dev_id – the device on which the function should be executed.

• stream_id – the CUDA stream on which the operation should be executed.

• cublas – the CublasCaller object that manages the cuBLAS handle.

api/function_LinearAlg_8hpp_1abee6aba9ad3ff05e525eee55e0f6553d

 

Download Python script

 

Download Notebook

 

View on GitHub