Поддержка Numba для объявления локальной памяти на устройстве AMD?
У меня есть код, который изначально был написан для работы с CUDA и использует локальную память, объявленную так:
numba.cuda.local.array(shape, type)
что здорово.
Я сейчас портирую этот код в ROCm, и я не вижу здесь эквивалента в документации Numba:
https://numba.pydata.org/numba-doc/latest/roc/memory.html
Но потом я заметил документ для функции numba.rocm.barrierговорит: "... при доступе к глобальной или локальной памяти соответственно". - ссылка на локальную память.
Итак, мой вопрос: есть ли способ объявить локальный массив с Numba на rocm, похожий на cuda.local.array?
Спасибо!