Merge pull request #2188 from ROCmSoftwarePlatform/rocm-enable-fusion…

…-matmul-test [ROCm]: Re-enable matmul fusion test on ROCm
ROCm · Aug 14, 2023 · 81b9007 · 81b9007
2 parents ef05923 + 059fd07
commit 81b9007
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/tensorflow/core/kernels/matmul_op_test.cc b/tensorflow/core/kernels/matmul_op_test.cc
@@ -185,10 +185,6 @@ class FusedMatMulOpTest : public OpsTestBase {
                      .Attr("transpose_b", transpose_b)
                      .Finalize(&fused_matmul));
 
-#if !(GOOGLE_CUDA || TENSORFLOW_USE_ROCM)
-    printf("Disallowing the FusedMatMul GPU test (neither CUDA nor ROCM observed)\n");
-    allow_gpu_device = false;
-#endif
     RunAndFetch(root, fused_matmul.name(), output, allow_gpu_device,
                 &fused_matmul);
   }