Есть ли способ, которым мы могли бы одновременно запускать функции на CPU и GPU (используя Python)? Я уже использую Numba для составления расписаний на уровне потоков для сложных вычислительных функций на GPU, но теперь мне нужно добавить параллелизм между CPU-GPU. Как только мы убедимся, что в общей памяти GPU есть все данные для начала обработки, мне нужно запустить GPU, а затем параллельно запустить некоторые функции на хосте с использованием CPU.
Я уверен, что время, затрачиваемое графическим процессором на возврат данных, намного больше, чем процессором для завершения задачи. Так что после того, как графический процессор завершил обработку, процессор уже ожидает загрузки данных на хост. Есть ли стандартная библиотека / способ достижения этого? Цените любые указания на этот счет.