Skip to content

Commit d51a825

Browse files
committed
Add warning about getri not supported by cuSolver
1 parent 238e2c6 commit d51a825

File tree

1 file changed

+3
-0
lines changed
  • source/module_base/module_container/ATen/kernels

1 file changed

+3
-0
lines changed

source/module_base/module_container/ATen/kernels/lapack.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -184,6 +184,9 @@ struct lapack_getrf {
184184
* @brief Computes the inverse of a matrix using LAPACK's GETRI routine.
185185
*
186186
* GETRI computes the inverse of a matrix using the LU factorization computed by GETRF.
187+
*
188+
* @warning cuSOLVER does \b not provide LU-based matrix inversion interface (getri).
189+
* To compute the inverse on \b GPU, use getrs instead.
187190
*/
188191
template <typename T, typename Device>
189192
struct lapack_getri {

0 commit comments

Comments
 (0)