Я имею в виду использовать getrf и getrs из пакета cuSolver и решить AB=X с помощью B=I.
Это самый лучший способ решить эту проблему?
Если да, то как лучше всего создать идентификационную матрицу col-major B в памяти устройства? Это можно сделать тривиально, используя цикл for, но это 1. займет много памяти и 2. будет довольно медленным. Есть ли более быстрый способ?
Обратите внимание, что cuSolver, к сожалению, не предоставляет getri. Поэтому я должен использовать getrs.
Пока CUDA не предоставит LAPACK API getri, я думаю, что getrf и getrs — лучший выбор для инверсии больших матриц.
Матрица B имеет тот же размер, что и A, поэтому я не думаю, что выделение B заставляет эту задачу потреблять намного больше памяти, чем ее входные/выходные данные.
Сложность getrf и getrs равна O(n^3) и O(n^2) соответственно, а настройка B=I равна O(n^2) + O(n). Я не думаю, что это должно быть узким местом всей процедуры. Вы можете поделиться своей реализацией, чтобы мы могли проверить, в чем может быть проблема.
