Skip to content

Commit 930f02c

Browse files
MoFHekarhdong
authored andcommitted
[fix] HvdAllToAllEmbedding init with GPU device couldn't be ran in environment withou GPU.
Should not skip any Test when it ran successfully.
1 parent 4ff09dd commit 930f02c

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

tensorflow_recommenders_addons/dynamic_embedding/python/kernel_tests/horovod_sync_train_test.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -93,8 +93,6 @@ def test_all_to_all_embedding_trainable(self):
9393
if (tf.__version__ == "2.11.0" or tf.__version__ == "2.11.1"):
9494
self.skipTest(
9595
"The save function doesn't work with TF 2.11, skip the test.")
96-
if not is_gpu_available:
97-
self.skipTest('Only test when gpu is available.')
9896
if (is_macos() and is_arm64()):
9997
self.skipTest(
10098
"Apple silicon devices don't support synchronous training based on Horovod."
@@ -394,7 +392,6 @@ def __init__(self, init, dynamic=False):
394392
kv_creator = de.CuckooHashTableCreator(saver=de.FileSystemSaver(
395393
proc_size=hvd.size(), proc_rank=hvd.rank()))
396394
self.emb = de.keras.layers.HvdAllToAllEmbedding(embedding_size=dim,
397-
devices=['/GPU:0'],
398395
initializer=0,
399396
kv_creator=kv_creator,
400397
name=name)

0 commit comments

Comments
 (0)