Skip to content

GitLab

Commit 07c09ccc authored by Igor Saprykin's avatar Igor Saprykin Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 263669433
parent 47e783e6
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/transformer/v2/transformer_main_test.py
View file @ 07c09ccc
...@@ -103,7 +103,8 @@ class TransformerTaskTest(tf.test.TestCase): ...@@ -103,7 +103,8 @@ class TransformerTaskTest(tf.test.TestCase):
t = tm.TransformerTask(FLAGS) t = tm.TransformerTask(FLAGS)
t.train() t.train()
@unittest.skipUnless(tf.test.is_built_with_cuda(), 'requires GPU') @unittest.skipIf(tf.test.is_built_with_cuda(), 'TODO(b/139497127): '
'Test is skipped because tf.pad doesn\'t work with GPU.')
def test_train_fp16(self): def test_train_fp16(self):
FLAGS.dtype = 'fp16' FLAGS.dtype = 'fp16'
t = tm.TransformerTask(FLAGS) t = tm.TransformerTask(FLAGS)
......
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