通過CUBLAS或THRUST在CUDA編程bsxfun？

Question

我有一個向量V有nx1項目和矩陣M有nxm項目。我想用CUDA將M的所有列與V相加。 THRUST或CUBLAS中是否有任何方法可以幫助我解決問題？

Answer 1

在這裏，我有兩個向下票

thrust::device_vector<float> dummy_y(N,1); 
alpha = 1.0f; 
beta = 0.0f; 
stat = cublasSger_v2(handle,K,N,&alpha,thrust::raw_pointer_cast(&dummy_y[0]),1,thrust::raw_pointer_cast(&A_sum_vec_d[0]),1,thrust::raw_pointer_cast(&C_d[0]),K); 

我通過行向量，滿1秒乘以複製的列向量，如矩陣的大小和cublasSger總結做什麼。 stat保持代碼的執行狀態，您可以檢查是否有任何可能的錯誤。