Add XLA test (#9848)

2021-01-29 11:25:03 +01:00
parent 99b9affa02
commit fdcde144d8
18 changed files with 85 additions and 0 deletions
--- a/tests/test_modeling_tf_bart.py
+++ b/tests/test_modeling_tf_bart.py
@@ -281,6 +281,10 @@ class TFBartModelTest(TFModelTesterMixin, unittest.TestCase):
        # TODO JP: Make BART float16 compliant
        pass

+    def test_xla_mode(self):
+        # TODO JP: Make BART XLA compliant
+        pass
+

 def _assert_tensors_equal(a, b, atol=1e-12, prefix=""):
    """If tensors not close, or a and b arent both tensors, raise a nice Assertion error."""