Можно ли читать / писать память 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;
    }

на самом деле этот код вызовет coredump в моей настройке. у меня вопрос: можно ли таким образом инициализировать bufferTokens? если это правда, как я могу позволить ему работать хорошо?

0 ответов

Другие вопросы по тегам