微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

可以直接读/写AMD GPU内存吗?

如何解决可以直接读/写AMD GPU内存吗?

在 roc_shmem 开源中: https://github.com/ROCm-Developer-Tools/ROC_SHMEM

./library/src/gpu_ib/backend.cpp:51

代码如下:

    CHECK_HIP(hipMalloc((void**) &bufferTokens,sizeof(unsigned int) * num_wgs));

    for (int i = 0; i < num_wgs; i++) {
        bufferTokens[i] = 0;
    }

实际上这段代码会在我的设置中导致核心转储。 我的问题是可以像这样初始化“bufferTokens”吗?如果是真的,我怎样才能让它运行良好?

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。