Disable test that invokes rocBlas TRSM

bmzhao · Jan 28, 2020 · 1f4186c · 1f4186c
1 parent e1a66cf
commit 1f4186c
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/tensorflow/compiler/xla/tests/triangular_solve_test.cc b/tensorflow/compiler/xla/tests/triangular_solve_test.cc
@@ -349,7 +349,11 @@ XLA_TEST_F(TriangularSolveTest, SimpleLeftUpperNotransposeUnitDiagonal) {
                              ErrorSpec(1e-2, 1e-2));
 }
 
-XLA_TEST_F(TriangularSolveTest, SimpleRightLowerTransposeConjugate) {
+// The following test will results in a call to "BlasTrsm".
+// That operation is currently not supported for the complex type on the ROCm
+// platform.
+XLA_TEST_F(TriangularSolveTest,
+           DISABLED_ON_GPU_ROCM(SimpleRightLowerTransposeConjugate)) {
   XlaBuilder builder(TestName());
 
   XlaOp a, b;
@@ -375,7 +379,11 @@ XLA_TEST_F(TriangularSolveTest, SimpleRightLowerTransposeConjugate) {
       &builder, expected, {a_data.get(), b_data.get()}, ErrorSpec(1e-2, 1e-2));
 }
 
-XLA_TEST_F(TriangularSolveTest, SimpleLeftUpperTransposeNoconjugate) {
+// The following test will results in a call to "BlasTrsm".
+// That operation is currently not supported for the complex type on the ROCm
+// platform.
+XLA_TEST_F(TriangularSolveTest,
+           DISABLED_ON_GPU_ROCM(SimpleLeftUpperTransposeNoconjugate)) {
   XlaBuilder builder(TestName());
 
   XlaOp a, b;