experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[ROCm] Fix for a bug in the ROCm implementation for matrix_triangular_solve op

Browse Source
This commit is contained in:
Deven Desai 2020-02-12 15:14:49 +00:00
parent 675799d105
commit b5a3eda2dc
1 changed files with 1 additions and 1 deletions

2
tensorflow/core/kernels/matrix_triangular_solve_op_impl.h
View File

@ -332,7 +332,7 @@ struct LaunchBatchMatrixTriangularSolve<GPUDevice, Scalar> {
// output' = rhs' / matrix' (' stands for transpose) // output' = rhs' / matrix' (' stands for transpose)
// Upper/lower needs to be swapped for this. // Upper/lower needs to be swapped for this.
uplo = lower ? rocblas_fill_upper : rocblas_fill_upper; uplo = lower ? rocblas_fill_upper : rocblas_fill_lower;
trans = adjoint ? rocblas_operation_conjugate_transpose trans = adjoint ? rocblas_operation_conjugate_transpose
: rocblas_operation_none; : rocblas_operation_none;
auto solver = absl::make_unique<ROCmSolver>(context); auto solver = absl::make_unique<ROCmSolver>(context);