cuda-将设备的指针拷贝到同一个设备的指针

tensorRT推理时,输入输出指针都位于设备上,所以设备上的指针之间的拷贝需要使用cuda的拷贝指令,不能使用主机的拷贝指令。

    float* fusion_model_feature_input1;
    fusion_model_feature_input1 = static_cast<float*>(fusion_model_trt_ptr_->getBuffer(system_param_.fusion_model_input_name1));
    CHECK_CUDA_ERROR(cudaMemsetAsync(fusion_model_feature_input1, 0, (system_param_.bev_shape[0] / 2) * (system_param_.bev_shape[1] / 2) * 64 * sizeof(float), stream_));
    CHECK_CUDA_ERROR(cudaMemcpy(fusion_model_feature_input1, bev_model_feature_output1, (system_param_.bev_shape[0] / 2) * (system_param_.bev_shape[1] / 2) * 64 * sizeof(float), cudaMemcpyDeviceToDevice));

上述代码中即是将cuda中的bev_model_feature_output1指针拷贝到cuda中的fusion_model_feature_input1指针中

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值