CUDA实现高效查找--审核未通过?

#include <stdio.h>
#include <vector>
#include <iostream>

using namespace std;

void __global__ bisearch(int *d, int *dval, int *pos, const int N);

void __global__ bisearch(int *d, int *dval, int *pos, const int N)
{

    int tid = blockDim.x * blockIdx.x + threadIdx.x;
    if(tid < N)
    {
       if(d[tid] == *dval)
       {*pos = tid;}
    }
}


int main(int argc, char *arcv[])

{
    if(argc<3)
    {
        printf("assignment is not enough!");
    }

    int N = atoi(arcv[1]);
    int val = atoi(arcv[2]);

    // host memory
    int hpos = -1;
    int a[N] = {0,1,2};
    vector<int> vec(a,a+N);
    
    /*  
    for(int i=0; i<N; i++)
    {
        vec.push_back(i);
    }

    cout<< vec[0] <<endl;
    cout<< vec[1] <<endl;
    cout<< vec[2] << endl;
    cout<< val<<endl;
    */

    int *h;
    h = (int *)&vec[0];
    
    // device memory
    int *d;
    cudaMalloc(&d, sizeof(int)*N);
    int *dval;
    cudaMalloc(&dval, sizeof(int));
    int *dpos;
    cudaMalloc(&dpos, sizeof(int));


    // host --> device
    cudaMemcpy( d, h, sizeof(int)*N,cudaMemcpyHostToDevice);
    cudaMemcpy( dval, &val, sizeof(int), cudaMemcpyHostToDevice);
    cudaMemcpy( dpos, &hpos, sizeof(int), cudaMemcpyHostToDevice);

    // threads and blocks

    const int block_size = 128;
    const int grid_size = (N + block_size -1) / block_size;

    bisearch<<<grid_size,block_size>>>(d, dval, dpos, N);

    // cuda to cpu
    cudaMemcpy( &hpos, dpos, sizeof(int), cudaMemcpyDeviceToHost   );
    



    printf("Ans: %d\n", hpos);


    // free memory
    cudaFree(d);
    cudaFree(dval);
    cudaFree(dpos);

    return 0;
    

}

总结

  vector不太会用;vector定义需要使用using namespace std;

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值