Как утекать графические графические услуги решателей PDE в Python? ⇐ Python

[Anonymous]

Я работал над небольшим пакетом Python, чтобы решить класс PDE с использованием scipy.integrate.solve_ivp . Поскольку дискретизации становятся все более тонкими, время выполнения становится узким местом, особенно когда мне нужно решить PDE для большого количества различных начальных условий. вещи, и я планирую, как лучше всего интегрировать вычисления на основе графических процессоров в мою текущую реализацию. solve_ivp , но сделайте данные о передаче функции ODE в графический процессор, выполните прямой шаг с чем -то вроде Cupy, а затем перенесите обратно в ЦП? Или трансферные накладные расходы будут слишком большими для практической пользы? Переменный шаг для всех решений параллельно, а затем изменить назад. Будет ли это работать или это станет столкнуться с адаптивными методами пособия? ) способ использования ускорения графического процессора (предпочтительно с использованием python)? ценится.

Подробнее здесь: https://stackoverflow.com/questions/794 ... -in-python