Skip to content

GitLab

Commit 32235d83 authored by Igor Saprykin's avatar Igor Saprykin Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 265499806
parent 12f9403f
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
official/transformer/v2/transformer_main_test.py
View file @ 32235d83
...@@ -20,6 +20,7 @@ from __future__ import print_function ...@@ -20,6 +20,7 @@ from __future__ import print_function
import os import os
import re import re
import sys
import unittest import unittest
from absl import flags from absl import flags
...@@ -178,6 +179,8 @@ class TransformerTaskTest(tf.test.TestCase): ...@@ -178,6 +179,8 @@ class TransformerTaskTest(tf.test.TestCase):
def test_eval(self): def test_eval(self):
if context.num_gpus() >= 2: if context.num_gpus() >= 2:
self.skipTest('No need to test 2+ GPUs without a distribution strategy.') self.skipTest('No need to test 2+ GPUs without a distribution strategy.')
if 'test_xla' in sys.argv[0]:
self.skipTest('TODO(xla): Make this test faster under XLA.')
self._prepare_files_and_flags() self._prepare_files_and_flags()
t = tm.TransformerTask(FLAGS) t = tm.TransformerTask(FLAGS)
t.eval() t.eval()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment