微凉的衣柜 2021-05-25 00:10 采纳率: 62.9%
浏览 25
已结题

如何使用 packed_accessor64 访问tensor上的数据?

参考了https://blog.csdn.net/weixin_43742643/article/details/116307036

和官方文档https://pytorch.org/cppdocs/notes/tensor_basics.html的访问cuda上tensor的代码,还是没有跑通:

__global__ void packed_accessor_kernel(
    PackedTensorAccessor64<float, 2> foo,
    float* trace) {
  int i=threadIdx.x
  gpuAtomicAdd(trace, foo[i][i])
}
 
torch::Tensor foo = torch::rand({12, 12});
 
// assert foo is 2-dimensional and holds floats.
auto foo_a = foo.packed_accessor64<float,2>();
float trace = 0;
 
packed_accessor_kernel<<<1, 12>>>(foo_a, &trace);

有没有哪位大佬跑通了这段代码,麻烦给个示例,谢谢了!

  • 写回答

1条回答 默认 最新

  • 微凉的衣柜 2021-09-07 16:47
    关注

    这段代码是cuda编程的代码。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 系统已结题 9月15日
  • 已采纳回答 9月7日

悬赏问题

  • ¥15 visionmaster启动失败,提示为“机器不满足授权而被禁用”
  • ¥50 用logisim设计16位单时钟周期cpu
  • ¥15 IDEA中圈复杂度如何具体设置
  • ¥50 labview采集不了数据
  • ¥15 请上面代码做什么处理或什么混淆
  • ¥15 英雄联盟自定义房间置顶
  • ¥15 W5500网线插上无反应
  • ¥15 如何用字典的Key,显示在WPF的xaml中
  • ¥15 weautomate读取Excel表格信息然后填写到网页一直报错,如何解决?
  • ¥15 C#如何在Webview2中获取网页验证码