From b5a3eda2dc4b16b61467ef4afd5f8211681133b3 Mon Sep 17 00:00:00 2001 From: Deven Desai Date: Wed, 12 Feb 2020 15:14:49 +0000 Subject: [PATCH] [ROCm] Fix for a bug in the ROCm implementation for matrix_triangular_solve op --- tensorflow/core/kernels/matrix_triangular_solve_op_impl.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/core/kernels/matrix_triangular_solve_op_impl.h b/tensorflow/core/kernels/matrix_triangular_solve_op_impl.h index 48f2eec11a6..fb7e6f0f5ff 100644 --- a/tensorflow/core/kernels/matrix_triangular_solve_op_impl.h +++ b/tensorflow/core/kernels/matrix_triangular_solve_op_impl.h @@ -332,7 +332,7 @@ struct LaunchBatchMatrixTriangularSolve { // output' = rhs' / matrix' (' stands for transpose) // 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 : rocblas_operation_none; auto solver = absl::make_unique(context);