commit 9ff3e492c0b8ea62d5046dbb56890bb33a06743c
author George Karpenkov <cheshire@google.com> Tue Jun 29 14:55:00 2021 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Tue Jun 29 15:08:50 2021 -0700
tree 4e6b77e5d4f85a785201ea62a26ff14e153b8d74
parent 712d5988f48d7c07d8d29ca23a5ac88274ad0596

No need to filter XLA for int32/int64 convolutions

PiperOrigin-RevId: 382171555
Change-Id: I131a297cf9b48928d04789afab7ba90b13d751ee

tensorflow/python/kernel_tests/matmul_op_test.py

diff --git a/tensorflow/python/kernel_tests/matmul_op_test.py b/tensorflow/python/kernel_tests/matmul_op_test.py
index 1c36088..c59aae2 100644
--- a/tensorflow/python/kernel_tests/matmul_op_test.py
+++ b/tensorflow/python/kernel_tests/matmul_op_test.py
@@ -259,11 +259,6 @@
   # TF2 does not support placeholders under eager so we skip it
   for use_static_shape in set([True, tf2.enabled()]):
     for dtype in dtypes_to_test:
-      if test_util.is_xla_enabled() and (dtype == np.int32 or
-                                         dtype == np.int64):
-        # TODO(b/171924639): Enable this test when XLA DOT supports
-        # integer types.
-        continue
       for m in sizes:
         for n in sizes:
           for k in sizes: